Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

PyTorch

PyTorch is an open source machine learning library based on the Torch library, used for applications such as computer vision and natural language processing, primarily developed by Facebook’s AI Research lab.

https://github.com/XuyangBai/PPF-FoldNet

PyTorch reimplementation for "PPF-FoldNet: Unsupervised Learning of Rotation Invariant 3D Local Descriptors" https://arxiv.org/abs/1808.10322

point-cloud ppf-foldnet ppfnet pytorch registration

Last synced: 31 Jul 2024

https://github.com/osai-ai/dokai

Collection of Docker images for ML/DL and video processing projects

cuda deep-learning docker docker-image ffmpeg opencv python pytorch tensorrt video-processing

Last synced: 01 Aug 2024

https://github.com/xudejing/video-clip-order-prediction

Self-supervised Spatiotemporal Learning via Video Clip Order Prediction

cvpr2019 pytorch self-supervised-learning spatiotemporal video

Last synced: 02 Aug 2024

https://github.com/lonePatient/BERT-chinese-text-classification-pytorch

This repo contains a PyTorch implementation of a pretrained BERT model for text classification.

bert chinese chinese-text-classification nlp pytorch text-classification

Last synced: 01 Aug 2024

https://github.com/ritchieng/eigenvectors-from-eigenvalues

PyTorch implementation comparison of old and new method of determining eigenvectors from eigenvalues.

colab determining-eigenvectors eigenvalues eigenvectors pytorch

Last synced: 30 Jul 2024

https://github.com/mzaradzki/factorization-machine-for-prediction

Factorization Machine for regression and classification

keras machine-learning python pytorch theano

Last synced: 10 Aug 2024

https://github.com/paganpasta/eqxvision

A Python package of computer vision models for the Equinox ecosystem.

equinox python pytorch vision

Last synced: 01 Aug 2024

https://github.com/openvinotoolkit/openvino_contrib

Repository for OpenVINO's extra modules

arm inference-engine java nvidia-gpu openvino pytorch

Last synced: 31 Jul 2024

https://github.com/anibali/margipose

3D monocular human pose estimation

deep-learning pose-estimation pytorch

Last synced: 08 Aug 2024

https://github.com/xuarehere/yolo_series_deepsort_pytorch

Deepsort with yolo series. This project support the existing yolo detection model algorithm (YOLOV8, YOLOV7, YOLOV6, YOLOV5, YOLOV4Scaled, YOLOV4, YOLOv3', PPYOLOE, YOLOR, YOLOX ).

deep-sort deep-sort-tracking deepsort mot mot-tracking object-tracking pytorch sort yolo yolov3 yolov4 yolov4-deepsort yolov5 yolov5-deepsort yolov5-deepsort-pytorch yolov6 yolov6-deepsort yolov7 yolov7-deepsort yolov8-deepsort

Last synced: 02 Aug 2024

https://github.com/ajhamdi/MVTN

pytorch implementation of the ICCV'21 paper "MVTN: Multi-View Transformation Network for 3D Shape Recognition"

3d 3d-models classification deep-learning iccv2021 point-cloud pytorch

Last synced: 31 Jul 2024

https://github.com/chail/gan-ensembling

Invert and perturb GAN images for test-time ensembling

generative-adversarial-network image-classfication image-synthesis pytorch

Last synced: 03 Aug 2024

https://github.com/kunhe/FastAP-metric-learning

Code for CVPR 2019 paper "Deep Metric Learning to Rank"

cvpr deep-metric-learning learning-to-rank matconvnet metric-learning pytorch

Last synced: 03 Aug 2024

https://github.com/JinheonBaek/GMT

Official Code Repository for the paper "Accurate Learning of Graph Representations with Graph Multiset Pooling" (ICLR 2021)

graph graph-classification graph-multiset-pooling graph-multiset-transformer graph-neural-networks graph-pooling graph-reconstruction graph-representation graph-representation-learning graph-transformer pytorch

Last synced: 08 Aug 2024

https://github.com/chenyuntc/scene-baseline

PyTorch baseline for AI challenger Scene classification

image-classification pytorch

Last synced: 07 Aug 2024

https://github.com/slydg/gru_to_predict_timeseries

使用pytorch搭建的循环神经网络在股票数据时间序列上的应用

gru python3 pytorch sklearn

Last synced: 24 Sep 2024

https://github.com/eric-ai-lab/PEViT

Official implementation of AAAI 2023 paper "Parameter-efficient Model Adaptation for Vision Transformers"

fine-tuning image-classification parameter-efficient-tuning pytorch vision-transformer

Last synced: 01 Aug 2024

https://github.com/jmisilo/clip-gpt-captioning

CLIPxGPT Captioner is Image Captioning Model based on OpenAI's CLIP and GPT-2.

computer-vision cv deep-learning image-caption image-caption-generator image-captioning machine-learning nlp python pytorch

Last synced: 01 Aug 2024

https://github.com/doocs/deep-learning

🙃 深度学习实践与知识总结

ai deep-learning deep-neural-networks doocs pytorch tensorflow

Last synced: 31 Jul 2024

https://github.com/nitin-rathi/hybrid-snn-conversion

Training spiking networks with hybrid ann-snn conversion and spike-based backpropagation

ann-snn-conversion backpropagation-algorithm deep-learning pytorch snn spiking-neural-networks

Last synced: 02 Aug 2024

https://github.com/eveningglow/multitask-CycleGAN

Pytorch implementation of multitask CycleGAN with auxiliary classification loss

computer-vision cyclegan deep-learning gan generative-adversarial-network image-translation pytorch

Last synced: 03 Aug 2024

https://github.com/jkoutsikakis/pytorch-wrapper

Provides a systematic and extensible way to build, train, evaluate, and tune deep learning models using PyTorch.

data-science deep-learning machine-learning neural-network python pytorch pytorch-wrapper tensor

Last synced: 07 Aug 2024

https://github.com/takuseno/minerva

An out-of-the-box GUI tool for offline deep reinforcement learning

deep-learning deep-reinforcement-learning offline-rl pytorch

Last synced: 02 Aug 2024

https://github.com/wchliao/multi-task-image-classification

Multi-task learning for image classification implemented in PyTorch.

cifar-100 cifar10 deep-learning machine-learning multi-task-learning omniglot pytorch

Last synced: 03 Aug 2024

https://github.com/xunge/pytorch_lmdb_imagenet

Use lmdb to speed up imagenet dataset

imagenet lmdb pytorch

Last synced: 03 Aug 2024

https://github.com/hpcaitech/CachedEmbedding

A memory efficient DLRM training solution using ColossalAI

colossal-ai deep-learning dlrm embeddings nlp pytorch recommandation-system

Last synced: 01 Aug 2024

https://github.com/zhiyiYo/Alpha-Gobang-Zero

A gobang robot based on reinforcement learning.

alphazero gobang pyqt5 pytorch reinforcement-learning

Last synced: 02 Aug 2024

https://github.com/OxWearables/ssl-wearables

Self-supervised learning for wearables using the UK-Biobank (>700,000 person-days)

accelerometer deep-learning foundation-models human-activity-recognition pytorch self-supervised-learning wearable

Last synced: 01 Aug 2024

https://github.com/sophiefy/StellaVoiceChanger

Deep-learning-based voice changer, supporting local inference.

deep-learning python pytorch voice-conversion

Last synced: 03 Aug 2024

https://github.com/huangtinglin/MixGCF

MixGCF: An Improved Training Method for Graph Neural Network-based Recommender Systems, KDD2021

graph-neural-network information-retrieval negative-sampling network-embedding pytorch recommender-system

Last synced: 08 Aug 2024

https://github.com/lxc86739795/human_vehicle_parsing_platform

A pytorch codebase for human parsing and vehicle parsing

codebase human parsing person pytorch segmentation vehicle vehicleid veri-wild veri776

Last synced: 01 Aug 2024

https://github.com/Torsion-Audio/nn-inference-template

Neural network inference template for real-time cricital audio environments - presented at ADC23

audio audio-plugin deep-learning inference inference-engine juce libtorch machine-learning neural-network onnx onnxruntime pytorch tensorflow tensorflow-lite

Last synced: 03 Aug 2024

https://github.com/LBM-EPFL/PeSTo

Geometric deep learning method to predict protein binding interfaces from a protein structure.

deep-learning geometric-deep-learning protein-binding-site protein-protein-interface protein-structure pytorch

Last synced: 02 Aug 2024

https://github.com/roman-vygon/triplet_loss_kws

Learning Efficient Representations for Keyword Spotting with Triplet Loss

deep-learning keyword-spotting pytorch speech-recognition

Last synced: 01 Aug 2024

https://github.com/drivendataorg/zamba

A Python package for identifying 42 kinds of animals, training custom models, and estimating distance from camera trap videos

animals camera-traps chimps cli conservation deep-learning ecology gpu jungle machine-learning neural-network python pytorch pytorch-lightning video-processing videos

Last synced: 01 Aug 2024

https://github.com/alldbi/SuperMix

Pytorch implementation of CVPR2021 paper: SuperMix: Supervising the Mixing Data Augmentation

augmentation cvpr2021 deep-learning distillation pytorch saliency-detection supervised

Last synced: 02 Aug 2024

https://github.com/Coderx7/SimpleNet

This repository contains the architectures, Models, logs, etc pertaining to the SimpleNet Paper (Lets keep it simple: Using simple architectures to outperform deeper architectures )

cnn-model cnn-pytorch convolutional-neural-networks deep-learning fast-net image-classification imagenet pytorch simplenet

Last synced: 01 Aug 2024

https://github.com/sberbank-ai-lab/pytorch-lifestream

A library built upon PyTorch for building embeddings on discrete event sequences using self-supervision

machine-learning pytorch self-supervised-learning

Last synced: 07 Aug 2024

https://barquerogerman.github.io/BeLFusion/

[ICCV2023] Official PyTorch Implementation of "BeLFusion: Latent Diffusion for Behavior-Driven Human Motion Prediction". ICCV 2023

belfusion ddim ddpm deep-learning diffusion generative-models human-motion-generation human-motion-prediction iccv2023 latent-diffusion ldm motion-forecasting motion-prediction pose-forecasting pose-prediction pytorch

Last synced: 02 Aug 2024

https://github.com/BarqueroGerman/BeLFusion

[ICCV2023] Official PyTorch Implementation of "BeLFusion: Latent Diffusion for Behavior-Driven Human Motion Prediction". ICCV 2023

belfusion ddim ddpm deep-learning diffusion generative-models human-motion-generation human-motion-prediction iccv2023 latent-diffusion ldm motion-forecasting motion-prediction pose-forecasting pose-prediction pytorch

Last synced: 02 Aug 2024

https://github.com/Haiyang-W/CAGroup3D

[NeurIPS2022] This is the official code of "CAGroup3D: Class-Aware Grouping for 3D Object Detection on Point Clouds".

3d-object-detection pytorch scannet-dataset sunrgbd

Last synced: 31 Jul 2024

https://github.com/rubenwiersma/hsn

Code for SIGGRAPH paper CNNs on Surfaces using Rotation-Equivariant Features

3d convolutional-neural-networks geometric-deep-learning pytorch pytorch-geometric siggraph

Last synced: 31 Jul 2024

https://github.com/MaxHalford/pytorch-resample

🎲 Iterable dataset resampling in PyTorch

imbalanced-learning oversampling pytorch resampling undersampling

Last synced: 03 Aug 2024

https://github.com/pcyin/pytorch_basic_nmt

A simple yet strong implementation of neural machine translation in pytorch

deep-learning natural-language-processing neural-machine-translation pytorch pytorch-implmention

Last synced: 31 Jul 2024

https://github.com/buxihuo/OW-YOLO

Detect known and unknown objects in the open world(具有区分已知与未知能力的全新检测器))

autolabel deep-learning object-detection open-world open-world-detection pytorch unknown yolo yolov5 yolov8

Last synced: 02 Aug 2024

https://github.com/autostronomy/astrophot

A fast, flexible, automated, and differentiable astronomical image 2D forward modelling tool for precise parallel multi-wavelength photometry

astronomy python pytorch science-research scientific-computing

Last synced: 24 Sep 2024

https://github.com/TuragaLab/DECODE

This is the official implementation of our publication "Deep learning enables fast and dense single-molecule localization with high accuracy" (Nature Methods)

deep-learning gpu high-density localization-microscopy microscopy pytorch smlm

Last synced: 03 Aug 2024

https://github.com/lorenmt/minimal-isaac-gym

A Minimal Example of Isaac Gym with DQN and PPO.

dqn isaac-gym ppo pytorch

Last synced: 01 Aug 2024

https://github.com/HaochengWan/PVT

PVT: Point-Voxel Transformer for 3D Deep Learning

kitti modelnet40 point-cloud point-cloud-processing pvt pytorch s3dis shapenet sota

Last synced: 31 Jul 2024

https://github.com/qq456cvb/PRIN

Pointwise Rotation-Invariant Network (AAAI 2020)

pytorch rotation-invariant-features segmentation

Last synced: 31 Jul 2024

https://github.com/abap34/ml-lecture

東京工業大学 traP Kaggle班「機械学習講習会」の資料

deeplearning education machine-learning pytorch textbook

Last synced: 01 Aug 2024

https://github.com/amrzhd/mriskullstripping

Developing a UNet3D model for accurate MRI skull stripping using the Calgary Campinas 359 dataset, enhancing neuroimaging preprocessing workflows.

biomedical-image-processing image-segmentation monai mri mri-images nibabel python pytorch unet unet-image-segmentation

Last synced: 24 Sep 2024

https://github.com/naity/image_tabular

Integrate image and tabular data for deep learning

deep-learning fastai image-classification pytorch tabular-data

Last synced: 03 Aug 2024

https://github.com/svishnu88/TGS-SaltNet

Kaggle | 21st place solution for TGS Salt Identification Challenge

computer-vision deep-learning fastai kaggle-competition pytorch

Last synced: 07 Aug 2024

https://github.com/project-monai/monai-deploy-app-sdk

MONAI Deploy App SDK offers a framework and associated tools to design, develop and verify AI-driven applications in the healthcare imaging domain.

ai deep-learning deploy dicom healthcare image-processing machine-learning medical-imaging ml ml-infrastructure ml-platform mlops model-deployment model-serving monai pipeline python pytorch workflow

Last synced: 24 Sep 2024

https://github.com/necla-ml/diff-jpeg

Official and maintained implementation of the paper "Differentiable JPEG: The Devil is in the Details" [WACV 2024].

differentiable-codec differentiable-jpeg image-codec image-compression jpeg pytorch ste surrogate wacv

Last synced: 24 Sep 2024

https://github.com/writer/fitbert

Use BERT to Fill in the Blanks

bert fill-in-the-blank nlp python pytorch

Last synced: 25 Sep 2024

https://github.com/huggingface/optimum-graphcore

Blazing fast training of 🤗 Transformers on Graphcore IPUs

fine-tuning graphcore machine-learning pytorch training transformers

Last synced: 01 Aug 2024

https://github.com/ai4iacts/hexagdly

Process hexagonally sampled data with PyTorch

deep-learning hexagonal-grids pytorch

Last synced: 01 Aug 2024

https://github.com/kefirski/pytorch_TDNN

Time Delayed NN implemented in pytorch

cnn deep-learning python pytorch tdnn

Last synced: 03 Aug 2024

https://github.com/acl21/deep-active-learning-pytorch

A PyTorch toolkit with 8 popular deep active learning query methods implemented.

deep-active-learning image-classification pytorch toolkit

Last synced: 01 Aug 2024

https://github.com/anshulpaigwar/Attentional-PointNet

Attentional-PointNet is Deep-Neural-Network architecture for 3D object detection in point clouds

3d-object-detection attention-mechanism autonomous-vehicles object-detection point-cloud pointnet pytorch pytorch-implmention

Last synced: 31 Jul 2024

https://github.com/intel/torch-ccl

oneCCL Bindings for Pytorch*

oneccl pytorch

Last synced: 31 Jul 2024

https://github.com/kefirski/pytorch_Highway

Highway network implemented in pytorch

highway-network python pytorch

Last synced: 03 Aug 2024

https://github.com/MachineLearningLifeScience/stochman

Algorithms for computations on random manifolds made easier

machine-learning manifold python pytorch riemannian-geometry

Last synced: 02 Aug 2024

https://github.com/lromul/argus

Lightweight library for training neural networks in PyTorch

deep-learning machine-learning neural-network python pytorch

Last synced: 24 Sep 2024

https://github.com/ndrplz/differentiable-renderer

Rastering algorithm to approximate the rendering of a 3D model silhouette in a fully differentiable way.

differentiable-rendering eccv-2018 python pytorch rendering rendering-engine tensorflow tensorflow-models

Last synced: 24 Sep 2024

https://github.com/hellloxiaotian/acnet

Asymmetric CNN for image super-resolution (IEEE Transactions on Systmes, Man, and Cybernetics: Systems 2021)

blind-super-resolution cnn cnn-pytorch deep-learning deep-network deep-neural-networks image-processing low-level-vision python2 python27 pytorch pytorch-cnn super-resolution

Last synced: 24 Sep 2024

https://github.com/imdeepmind/neuralpy

NeuralPy: A Keras like deep learning library works on top of PyTorch

data-science deep-learning keras library machine-learning neural-network neuralpy neuralpy-torch python pytorch

Last synced: 24 Sep 2024

https://github.com/guyyariv/TempoTokens

This repo contains the official PyTorch implementation of: Diverse and Aligned Audio-to-Video Generation via Text-to-Video Model Adaptation

ai-art audio-to-video audio-visual deep-learning diffusion-models generative-ai modelscope pytorch video-synthesis

Last synced: 05 Aug 2024

https://github.com/ajhamdi/mvtorch

a Pytorch library for multi-view 3D understanding and generation

3d 3d-deep-learning deep learning multi-view multi-view-geometry multi-view-learning nerf pytorch pytorch3d

Last synced: 31 Jul 2024

https://github.com/chongruo/pytorch-HED

An implementation of HED in pytorch

pytorch

Last synced: 03 Aug 2024

https://github.com/dengfenglai321/YOLOv4-Hat-detection

基于YOLOv4的安全帽佩戴检测

hat-detection pytorch yolov4

Last synced: 02 Aug 2024

https://github.com/val-iisc/sketch-parse

Code, demos and data for SketchParse (a neural network for sketch segmentation). Paper:

deep-learning pytorch semantic-segmentation sketches

Last synced: 12 Aug 2024

https://github.com/dwromero/wavelet_networks

Code repository of the paper "Wavelet Networks: Scale-Translation Equivariant Learning From Raw Time-Series, TMLR" https://arxiv.org/abs/2006.05259

cnns equivariance group-convolution pytorch time-series

Last synced: 02 Aug 2024

https://github.com/theos-ai/easy-yolov7

This a clean and easy-to-use implementation of YOLOv7 in PyTorch, made with ❤️ by Theos AI.

computer-vision deep-learning machine-learning neural-networks object-detection python pytorch yolov5 yolov7

Last synced: 02 Aug 2024