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: 12 Nov 2024

https://github.com/xunge/pytorch_lmdb_imagenet

Use lmdb to speed up imagenet dataset

imagenet lmdb pytorch

Last synced: 15 Nov 2024

https://github.com/orobix/visual-feature-attribution-using-wasserstein-gans-pytorch

Implementation of Visual Feature Attribution using Wasserstein GANs (VAGANs, https://arxiv.org/abs/1711.08998) in PyTorch

anomaly-detection cnn gan python pytorch vgan

Last synced: 11 Nov 2024

https://github.com/JunlinHan/YOCO

Code for You Only Cut Once: Boosting Data Augmentation with a Single Cut, ICML 2022.

cifar10 cifar100 classification computer-vision contrastive-learning data-augmentation data-augmentation-strategies data-augmentations icml imagenet imagenet-classifier instance-segmentation object-detection pytorch rain-removal super-resolution

Last synced: 23 Aug 2024

https://github.com/mratsim/arch-data-science

Archlinux PKGBUILDs for Data Science, Machine Learning, Deep Learning, NLP and Computer Vision

archlinux cuda cudnn data-science deep-learning lightgbm machine-learning mkl mxnet natural-language-processing natural-language-understanding nervana opencv package pandas pytorch scikit-learn spacy tensorflow xgboost

Last synced: 09 Nov 2024

https://github.com/lromul/ball-action-spotting

SoccerNet@CVPR | 1st place solution for Ball Action Spotting Challenge 2023

action-recognition action-spotting deep-learning football pytorch soccer soccernet video video-processing

Last synced: 01 Nov 2024

https://github.com/tri-ml/dgp

ML Dataset Governance Policy for Autonomous Vehicle Datasets

autonomous-driving data-management deep-learning machine-learning pytorch

Last synced: 13 Nov 2024

https://github.com/mertyg/vision-language-models-are-bows

Experiments and data for the paper "When and why vision-language models behave like bags-of-words, and what to do about it?" Oral @ ICLR 2023

blip clip compositionality multimodal pytorch vision-language

Last synced: 19 Nov 2024

https://github.com/svip-lab/hrnet-for-fashion-landmark-estimation.pytorch

[DeepFashion2 Challenge] Fashion Landmark Estimation with HRNet

fashion keypoint pytorch

Last synced: 18 Nov 2024

https://github.com/donydchen/fmpn-fer

😁[VCIP'19 Oral] Official PyTorch Implementation of Facial Motion Prior Networks for Facial Expression Recognition

expression facial-expression-recognition pytorch vcip-2019

Last synced: 09 Nov 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: 12 Nov 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: 03 Nov 2024

https://github.com/zllrunning/hand-detection.pytorch

FaceBoxes for hand detection in PyTorch

cnn egohands-dataset hand-detection object-detection pytorch

Last synced: 08 Nov 2024

https://github.com/linxueyuanstdio/pytorch-ddpm

500 行代码实现降噪扩散模型 DDPM，干净无依赖

classifier-free ddpm pytorch

Last synced: 19 Nov 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: 07 Nov 2024

https://github.com/senya-ashukha/quantile-regression-dqn-pytorch

A short and easy implementation of Quantile Regression DQN | Distributional Reinforcement Learning

pytorch qr-dqn quantile-regression-dqn reinforcement-learning

Last synced: 19 Nov 2024

https://github.com/thunlp/re-context-or-names

Bert-based models(BERT, MTB, CP) for relation extraction.

bert contrastive-learning pretrain pytorch relation-extraction

Last synced: 10 Nov 2024

https://github.com/kaylode/vehicle-counting

Vehicle counting using Pytorch

deepsort machine-learning object-detection object-tracking pytorch vehicle-counting vehicle-tracking yolov5

Last synced: 09 Nov 2024

https://github.com/nishnik/capsnet-pytorch

My attempt at implementing CapsNet from the paper Dynamic Routing Between Capsules

convolutional-neural-networks deep-learning pytorch

Last synced: 18 Nov 2024

https://github.com/epwalsh/nlp-models

NLP research experiments, built on PyTorch within the AllenNLP framework.

allennlp nlp pytorch pytorch-nlp

Last synced: 01 Nov 2024

https://github.com/thudm/grb

Graph Robustness Benchmark: A scalable, unified, modular, and reproducible benchmark for evaluating the adversarial robustness of Graph Machine Learning.

adversarial-attacks deep-learning graph-neural-networks machine-learning pytorch

Last synced: 14 Nov 2024

https://github.com/kamalkraj/stable-diffusion-tritonserver

Deploy stable diffusion model with onnx/tenorrt + tritonserver

deploy docker fp16 inference machine-learning nvidia onnx python3 pytorch stablediffusion tensorrt tensorrt-inference transformers triton-inference-server

Last synced: 07 Nov 2024

https://github.com/zhiyiyo/alpha-gobang-zero

A gobang robot based on reinforcement learning.

alphazero gobang pyqt5 pytorch reinforcement-learning

Last synced: 27 Oct 2024

https://github.com/sophiefy/StellaVoiceChanger

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

deep-learning python pytorch voice-conversion

Last synced: 18 Nov 2024

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

🎲 Iterable dataset resampling in PyTorch

imbalanced-learning oversampling pytorch resampling undersampling

Last synced: 02 Nov 2024

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

A gobang robot based on reinforcement learning.

alphazero gobang pyqt5 pytorch reinforcement-learning

Last synced: 14 Nov 2024

https://github.com/lonepatient/electra_pytorch

ELECTRA: Pre-training Text Encoders as Discriminators Rather Than Generators

bert deeplearning electra glue language-model nlp pytorch

Last synced: 06 Nov 2024

https://github.com/edvardlindelof/graph-neural-networks-for-drug-discovery

astrazeneca deep-learning drug-discovery graph-neural-networks pytorch

Last synced: 30 Oct 2024

https://github.com/jahongir7174/efficientnetv2

EfficientNetV2 implementation using PyTorch

efficientnetv2 imagenet pytorch training

Last synced: 18 Nov 2024

https://github.com/cedrickchee/data-science-notebooks

Data science Python notebooks—a collection of Jupyter notebooks on machine learning, deep learning, statistical inference, data analysis and visualization.

data-science deep-learning fastai kaggle keras machine-learning notebooks numpy pandas python pytorch tensorflow

Last synced: 17 Nov 2024

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

🎲 Iterable dataset resampling in PyTorch

imbalanced-learning oversampling pytorch resampling undersampling

Last synced: 15 Nov 2024

https://github.com/soulteary/docker-stable-diffusion-taiyi

带 Web UI 的 Docker taiyi (太乙)，中文 Stable Diffusion WebUI 一键运行环境。

docker pytorch stable-diffusion taiyi webui

Last synced: 02 Nov 2024

https://github.com/dagshub/client

DagsHub client libraries

ai data data-science data-streaming dvc hacktoberfest hacktoberfest2023 keras machine-learning machinelearning mlops python pytorch tensorflow

Last synced: 22 Oct 2024

https://github.com/sony/san

PyTorch implementation of slicing adversarial network (SAN)

deep-generative-model gan generative-model machine-learning optimal-transport pytorch sliced-wasserstein-distance wasserstein-distance

Last synced: 15 Nov 2024

https://github.com/kevinzakka/ibc

A PyTorch implementation of Implicit Behavioral Cloning

behavioral-cloning deep-learning energy-based-model pytorch robotics

Last synced: 28 Oct 2024

https://github.com/oaid/tengine-convert-tools

Tengine Convert Tool supports converting multi framworks' models into tmfile that suitable for Tengine-Lite AI framework.

artificial-intelligence cnn dnn mxnet onnx pytorch tensorflow

Last synced: 05 Nov 2024

https://github.com/d-li14/ghostnet.pytorch

73.6% GhostNet 1.0x pre-trained model on ImageNet

ghostnet imagenet mobilenetv3 pretrained-models pytorch reproduction

Last synced: 07 Nov 2024

https://github.com/masadcv/fastgeodis

Fast Implementation of Generalised Geodesic Distance Transform for CPU (OpenMP) and GPU (CUDA)

deep-learning gpu pytorch

Last synced: 14 Nov 2024

https://github.com/gle-bellier/flow-matching

Annotated Flow Matching paper

deep-learning flow-matching generative-ai machine-learning pytorch pytorch-lightning

Last synced: 16 Nov 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: 05 Nov 2024

https://github.com/polyaxon/mloperator

Machine learning operator & controller for Kubernetes

dask deep-learning k8s keras kubernetes kubernetes-operator machine-learning mlops mpi mxnet notebook pytorch scikit-learn spark tensorboard tensorflow xgboost

Last synced: 13 Nov 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: 08 Nov 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/young-geng/cql

Conservative Q Learning on top of SAC

pytorch reinforcement-learning

Last synced: 06 Nov 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: 14 Nov 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://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: 13 Nov 2024

https://github.com/wassname/attentive-neural-processes

implementing "recurrent attentive neural processes" to forecast power usage (w. LSTM baseline, MCDropout)

anp anp-rnn attention attentive-neural-processes neural-processes prediction pytorch rnn uncertainty

Last synced: 08 Nov 2024

https://github.com/lantiga/pytorch2c

A Python module for compiling PyTorch graphs to C

compiled-graphs deep-learning graph python pytorch tensor

Last synced: 22 Oct 2024

https://github.com/layerai-archive/sdk

Metadata store for Production ML

collaboration data-science data-versioning deep-learning experiment-tracking hyperparameter-optimization hyperparameter-tuning keras machine-learning mlops model-versioning python pytorch reinforcement-learning sklearn tensorflow

Last synced: 26 Sep 2024

https://github.com/mlabonne/graph-neural-network-course

Free hands-on course about Graph Neural Networks using PyTorch Geometric.

deep-learning graph-attention-networks graph-convolutional-networks graph-neural-networks neural-network pytorch pytorch-geometric

Last synced: 08 Nov 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: 07 Nov 2024

https://github.com/tomar840/two-stream-fusion-for-action-recognition-in-videos

action-recognition pretrained-models python pytorch two-stream-cnn ucf101

Last synced: 14 Nov 2024

https://github.com/amdegroot/pytorch-containers

Torch Containers simplified in PyTorch

autograd complex-architectures pytorch pytorch-containers torch

Last synced: 14 Nov 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: 09 Nov 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: 28 Oct 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: 10 Oct 2024

https://github.com/ashishpatel26/vision-transformer-keras-tensorflow-pytorch-examples

Tensorflow implementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformer

example keras pytorch tensorflow tensorflow2 vision-transformer

Last synced: 19 Nov 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: 02 Oct 2024

https://github.com/chrischoy/open-ucn

The first fully convolutional metric learning for geometric/semantic image correspondences.

correspondence fully-convolutional-networks image-analysis metric-learning pytorch

Last synced: 07 Nov 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: 11 Oct 2024

https://github.com/encounter1997/sfa

Official Implementation of "Exploring Sequence Feature Alignment for Domain Adaptive Detection Transformers"

detection-transformer domain-adaptation pytorch

Last synced: 14 Nov 2024

https://github.com/ahmedfgad/torchga

Train PyTorch Models using the Genetic Algorithm with PyGAD

cnn deep-learning evolutionary-algorithms genetic-algorithm machine-learning neural-networks neuroscience pygad python pytorch torch

Last synced: 17 Nov 2024

https://github.com/ailln/fuzi-nlu

🤖️ 聊天机器人——夫子的「自然语言理解」模块

chatbot intent-classification joint-nlu nlu pytorch question-answering seq2seq seq2seq-chatbot slot-filling socketio

Last synced: 18 Nov 2024

https://github.com/M-Nauta/ProtoTree

ProtoTrees: Neural Prototype Trees for Interpretable Fine-grained Image Recognition, published at CVPR2021

computer-vision cvpr2021 decision-trees deep-neural-networks explainability explainable-ai explainable-ml fine-grained-classification fine-grained-visual-categorization interpretability interpretable-deep-learning interpretable-machine-learning pytorch

Last synced: 15 Nov 2024

https://github.com/oxwhirl/treeqn

deep-learning deep-reinforcement-learning openai-gym pytorch reinforcement-learning

Last synced: 08 Nov 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: 30 Oct 2024

https://github.com/mberr/torch-ppr

(Personalized) Page-Rank computation using PyTorch

gpu pagerank personalized-pagerank pytorch

Last synced: 18 Nov 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: 06 Nov 2024

https://github.com/thunlp/prompt-transferability

On Transferability of Prompt Tuning for Natural Language Processing

nlp parameter-efficient-learning parameter-efficient-tuning pretrained-language-model pretrained-language-models pretrained-models prompt prompt-tuning pytorch transfer-learning

Last synced: 10 Nov 2024

https://github.com/fcakyon/balanced-loss

Easy to use class balanced cross entropy and focal loss implementation for Pytorch

balanced-loss binary-crossentropy class-balanced-loss computer-vision cross-entropy cvpr deep-learning focal-loss image-classification loss-functions machine-learning pip pypi python pytorch

Last synced: 15 Nov 2024

https://github.com/graykode/modelsummary

All Model summary in PyTorch similar to `model.summary()` in Keras

keras python pytorch summary

Last synced: 01 Nov 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: 03 Nov 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: 28 Oct 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: 04 Nov 2024

https://github.com/bshall/acoustic-model

Acoustic models for: A Comparison of Discrete and Soft Speech Units for Improved Voice Conversion

pytorch representation-learning speech voice-conversion

Last synced: 13 Nov 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: 15 Nov 2024

https://github.com/patrick-kidger/fasterneuraldiffeq

Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)

controlled-differential-equations deep-learning deep-neural-networks differential-equations dynamical-systems machine-learning neural-differential-equations numerical-analysis numerical-methods ordinary-differential-equations pytorch

Last synced: 08 Nov 2024

https://github.com/cxhernandez/molencoder

Molecular AutoEncoder in PyTorch

autoencoder computational-chemistry deep-learning python pytorch

Last synced: 27 Oct 2024

https://github.com/traveller59/torch2trt

convert torch module to tensorrt network or tvm function

pytorch tensorrt tvm

Last synced: 17 Nov 2024

https://github.com/interdigitalinc/compressai-vision

CompressAI-Vision helps you design, test and compare Video Compression for Machines pipelines. Compression methods can be either pulled from custom AI-based modules from CompressAI or traditional codecs such as H.266/VVC.

compression computer-vision deep-learning machine-to-machine-communication pytorch video-compression

Last synced: 14 Nov 2024

https://github.com/j-min/dropouts

PyTorch Implementations of Dropout Variants

bayesian-neural-networks dropout gaussian-dropout local-reparametrization-trick pytorch variational-dropout variational-inference

Last synced: 19 Nov 2024

https://github.com/ai4iacts/hexagdly

Process hexagonally sampled data with PyTorch

deep-learning hexagonal-grids pytorch

Last synced: 01 Nov 2024

https://github.com/comet-ml/issue-tracking

Questions, Help, and Issues for Comet ML

comet-ml keras machine-learning pytorch scikit-learn tenserflow

Last synced: 05 Nov 2024

https://github.com/rucaibox/recbole-cdr

This is a library built upon RecBole for cross-domain recommendation algorithms

cross-domain pytorch recbole recommendation

Last synced: 11 Oct 2024

https://github.com/ofnote/tryongan

TryOnGAN: Unofficial Implementation

deep-learning fashion gan generative-adversarial-network pytorch viton

Last synced: 16 Nov 2024

https://github.com/qq456cvb/PRIN

Pointwise Rotation-Invariant Network (AAAI 2020)

pytorch rotation-invariant-features segmentation

Last synced: 28 Oct 2024

https://github.com/hsankesara/voxelmorph-pytorch

An unofficial PyTorch implementation of VoxelMorph- An unsupervised 3D deformable image registration method

cvpr2018 deformable-image-registration ieeetran image-registration pytorch spatial-transformer-networks voxel-morph

Last synced: 27 Oct 2024

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

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

deeplearning education machine-learning pytorch textbook

Last synced: 05 Nov 2024

https://github.com/bjarten/computer-vision-nd

Projects and exercises for the Udacity Computer Vision Nanodegree

char cnn computer cvnd detection face facial image-captioning keypoints learning lstm machine nanodegree pytorch rnn slam tutorial udacity vision yolo

Last synced: 29 Oct 2024

https://github.com/scitator/run-skeleton-run

Reason8.ai PyTorch solution for NIPS RL 2017 challenge

actor-critic ddpg ddpg-agent nips nips-2017 physics-based ppo pytorch pytorch-solution reinforcement-learning tensorflow trpo

Last synced: 22 Oct 2024

https://github.com/unity-technologies/datasetinsights

Synthetic Dataset Insights

computer-vision machine-learning pytorch synthetic-datasets unity-simulation

Last synced: 07 Oct 2024

https://github.com/d-li14/dhm

[CVPR 2020] Dynamic Hierarchical Mimicking Towards Consistent Optimization Objectives

cvpr2020 deeply-supervised imagenet knowledge-distillation pretrained-models pytorch

Last synced: 07 Nov 2024

https://github.com/tohrusky/final2x-core

core for Final2x

artificial-intelligence ccrestoration computer-vision cross-platform image-processing low-level-vision ncnn opencv python3 pytorch super-resolution

Last synced: 11 Nov 2024

https://github.com/dusty-nv/jetson-voice

ASR/NLP/TTS deep learning inference library for NVIDIA Jetson using PyTorch and TensorRT

deep-learning jetson jetson-nano nlp pytorch speech-recognition tensorrt text-to-speech

Last synced: 07 Nov 2024

https://github.com/anthony-wang/CrabNet

Predict materials properties using only the composition information!

attention attention-mechanism machine-learning materials-informatics materials-science materials-screening neural-networks predict-materials-properties pytorch scikit-learn self-attention transformer

Last synced: 14 Nov 2024

https://github.com/rasbt/pytorch-memory-optim

This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog post.

deep-learning llm memory-optimization pytorch vision vit

Last synced: 01 Nov 2024

https://github.com/MachineLearningLifeScience/stochman

Algorithms for computations on random manifolds made easier

machine-learning manifold python pytorch riemannian-geometry

Last synced: 12 Nov 2024

https://github.com/eric-canas/qrdet

Robust QR Detector based on YOLOv8

easy-to-use object-detection pip pytorch qr-detection qrcode yolov8

Last synced: 13 Nov 2024

https://github.com/hkchengrex/scribble-to-mask

[CVPR 2021] MiVOS - Scribble to Mask module

computer-vision cvpr2021 deep-learning interactive-segmentation pytorch segmentation

Last synced: 10 Nov 2024