https://github.com/marload/aquvitae

Knowledge Distillation Toolkit

deep-learning knowledge-distillation light-weight machine-learning model-compression pytorch tensorflow

Last synced: 02 Dec 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/sebgao/ctensor

A super light-weight deep learning library based on NumPy in PyTorch fashion.

deep-learning deep-neural-networks numpy pytorch

Last synced: 26 Dec 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/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/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: 28 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: 15 Feb 2025

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/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: 01 Dec 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/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/lantiga/pytorch2c

A Python module for compiling PyTorch graphs to C

compiled-graphs deep-learning graph python pytorch tensor

Last synced: 06 Jan 2025

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: 18 Feb 2025

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/avinashpaliwal/deep-slomo

Official PyTorch implementation of "Deep Slow Motion Video Reconstruction with Hybrid Imaging System" (TPAMI)

convolutional-neural-networks deep-learning deep-slomo dual-cameras frame-interpolation hybrid-imaging pytorch pytorch-implementation slomo slow-motion super-resolution video-frame-interpolation video-super-resolution

Last synced: 05 Dec 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/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/amdegroot/pytorch-containers

Torch Containers simplified in PyTorch

autograd complex-architectures pytorch pytorch-containers torch

Last synced: 14 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/djsaunde/spiketorch

Experiments with spiking neural networks (SNNs) in PyTorch. See https://github.com/BINDS-LAB-UMASS/bindsnet for the successor to this project.

machine-learning neurons python pytorch snn spiking-neural-networks

Last synced: 24 Dec 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: 19 Jan 2025

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: 16 Feb 2025

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/snap-research/graphless-neural-networks

[ICLR 2022] Code for Graph-less Neural Networks: Teaching Old MLPs New Tricks via Distillation (GLNN)

deep-learning distillation efficient-inference gnn graph-algorithm graph-neural-networks knowledge-distillation pytorch scalability

Last synced: 13 Feb 2025

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 Feb 2025

https://github.com/bluemirrors/cvu

Computer Vision deployment tools for dummies and experts. CVU aims at making CV pipelines easier to build and consistent around platforms, devices, and models.

computer-vision object-detection onnxruntime pytorch tensorflow2 tensorrt yolov5

Last synced: 10 Feb 2025

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/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/lartpang/hdfnet

(ECCV 2020) Hierarchical Dynamic Filtering Network for RGB-D Salient Object Detection

eccv eccv2020 image-processing imageprocessing imagesegmentation pytorch saliency saliency-detection salient-object-detection segmentation

Last synced: 17 Jan 2025

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/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/oxwhirl/treeqn

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

Last synced: 08 Nov 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/mberr/torch-ppr

(Personalized) Page-Rank computation using PyTorch

gpu pagerank personalized-pagerank pytorch

Last synced: 18 Nov 2024

https://github.com/graykode/modelsummary

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

keras python pytorch summary

Last synced: 20 Dec 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/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/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/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/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/hzxie/rmnet

The official implementation of "Efficient Regional Memory Network for Video Object Segmentation". (Xie et al., CVPR 2021)

cvpr2021 pytorch segmentation video-object-segmentation

Last synced: 06 Dec 2024

https://github.com/r00tman/eventnerf

Neural Radiance Fields from a Single Colour Event Camera [CVPR 2023]

computer-vision cvpr2023 event-camera eventnerf mesh nerf neural-radiance-fields neural-rendering pytorch volume-rendering

Last synced: 09 Dec 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: 06 Dec 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/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/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/traveller59/torch2trt

convert torch module to tensorrt network or tvm function

pytorch tensorrt tvm

Last synced: 17 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/onnx/turnkeyml

The no-code AI toolchain

ai benchmark gpu llm npu nvidia onnx onnxruntime-genai pytorch toolchain x86

Last synced: 15 Feb 2025

https://github.com/edisonleeeee/greatx

A graph reliability toolbox based on PyTorch and PyTorch Geometric (PyG).

adversarial-attacks distribution-shift graph-convolutional-networks graph-neural-networks graph-reliability-toolbox inherent-noise pytorch pytorch-geometric

Last synced: 16 Feb 2025

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/filapro/unidet3d

[AAAI2025] UniDet3D: Multi-dataset Indoor 3D Object Detection

3d-object-detection mmdetection3d pytorch s3dis scannet

Last synced: 17 Feb 2025

https://github.com/qq456cvb/PRIN

Pointwise Rotation-Invariant Network (AAAI 2020)

pytorch rotation-invariant-features segmentation

Last synced: 28 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/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/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: 14 Feb 2025

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

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

deeplearning education machine-learning pytorch textbook

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

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/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/unity-technologies/datasetinsights

Synthetic Dataset Insights

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

Last synced: 27 Dec 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/yukkyo/pytorch-filterresponsenormalizationlayer

PyTorch implementation of "Filter Response Normalization Layer: Eliminating Batch Dependence in the Training of Deep Neural Networks"

batchnorm2d catalyst deep-neural-networks frn python pytorch

Last synced: 27 Jan 2025

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: 15 Feb 2025

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/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: 07 Jan 2025

https://github.com/Atharva-Phatak/torchflare

TorchFlare is a simple, beginner-friendly, and easy-to-use PyTorch Framework train your models effortlessly.

deep-learning deep-neural-networks gans generative-adversarial-network image-classfication image-classification image-processing image-segmentation infrastructure kaggle machine-learning natural-language-processing python pytorch pytorch-model self-supervised-learning text-classification

Last synced: 15 Nov 2024

https://github.com/dbaranchuk/memory-efficient-maml

Memory efficient MAML using gradient checkpointing

deep-learning gpu gradient-checkpointing maml meta-learning pytorch

Last synced: 07 Nov 2024

https://github.com/vitae-transformer/vitae-transformer-scene-text-detection

A comprehensive list [Hi-SAM@TPAMI'24, GoMatching@NeurIPS'24, DeepSolo(++)@ CVPR'23, DPText-DETR@AAAI'23, I3CL@IJCV'22] of our research works related to scene text detection, spotting, etc., including papers, codes.

deep-learning ocr pytorch scene-text-detection scene-text-spotting vision-transformer

Last synced: 17 Feb 2025

https://github.com/borhanMorphy/fastface

Light Face Detection using PyTorch Lightning

edge-ai facedetection pytorch pytorch-lightning

Last synced: 13 Jan 2025

https://github.com/neptune-ai/examples

📝 Examples of how to use Neptune for different use cases and with various MLOps tools

compare dashboard experiment fastai jupyter keras metadata ml mlops optuna pandas pipelines pytorch sklearn store tensorflow tracker tracking visualization xgboost

Last synced: 20 Feb 2025

https://github.com/daemon/pytorch-pcen

PyTorch reimplementation of per-channel energy normalization for audio.

audio pytorch speech

Last synced: 09 Dec 2024

https://github.com/d4l3k/torchdrive

torch models and primitives for birdseye view and voxel models for self driving tasks.

pytorch self-driving

Last synced: 18 Feb 2025

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

https://github.com/yuanzhoulvpi2017/quick_sentence_transformers

sentence-transformers to onnx 让sbert模型推理效率更快

bert pytorch sentence-transformers transformer transformers

Last synced: 08 Nov 2024

https://github.com/hellloxiaotian/deep-learning-on-image-denoising-an-overview

Deep Learning on Image Denoising: An overview (Neural Networks, 2020)

blind-denoising cnn denoising-network denoising-survery image-denoising keras keras-cnn low-level low-level-vision matconvnet python pytorch pytorch-cnn real-noise survery tensorflow tensorflow-cnn theano

Last synced: 11 Nov 2024

https://github.com/borhanmorphy/fastface

Light Face Detection using PyTorch Lightning

edge-ai facedetection pytorch pytorch-lightning

Last synced: 21 Jan 2025

https://github.com/Erlemar/wheat

Wheat Detection challenge on Kaggle

deep-learning hydra kaggle-competition object-detection pytorch pytorch-lightning torchvision

Last synced: 15 Nov 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: 09 Feb 2025

https://github.com/toshas/torch_truncnorm

Truncated Normal Distribution in PyTorch

distributions pytorch truncated-normal

Last synced: 25 Jan 2025

https://github.com/giotto-ai/giotto-deep

Deep learning made topological.

computational-geometry deep-learning image-processing nlp pytorch tda

Last synced: 19 Feb 2025

https://github.com/kefirski/pytorch_TDNN

Time Delayed NN implemented in pytorch

cnn deep-learning python pytorch tdnn

Last synced: 14 Nov 2024

https://github.com/lucasbotang/gradnorm

PyTorch implementation of the GradNorm

deep-learning gradnorm multitask-learning python pytorch

Last synced: 17 Feb 2025

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

Blazing fast training of 🤗 Transformers on Graphcore IPUs

fine-tuning graphcore machine-learning pytorch training transformers

Last synced: 02 Nov 2024

https://github.com/asteroid-team/asteroid-filterbanks

Asteroid's filterbanks :rocket:

asteroid-filterbanks audio audio-processing deep-learning filterbanks python3 pytorch

Last synced: 16 Feb 2025

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

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

基于YOLOv4的安全帽佩戴检测

hat-detection pytorch yolov4

Last synced: 09 Nov 2024

https://github.com/writer/fitbert

Use BERT to Fill in the Blanks

bert fill-in-the-blank nlp python pytorch

Last synced: 23 Jan 2025

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

oneCCL Bindings for Pytorch*

oneccl pytorch

Last synced: 30 Oct 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: 17 Feb 2025

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: 28 Oct 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: 05 Nov 2024

https://github.com/yusugomori/deeplearning-keras-tf2-torch

詳解ディープラーニング 第2版

deep-learning keras pytorch tensorflow2

Last synced: 07 Nov 2024

https://github.com/kevinzakka/form2fit

[ICRA 2020] Train generalizable policies for kit assembly with self-supervised dense correspondence learning.

deep-learning perception pytorch robotics self-supervised-learning

Last synced: 28 Oct 2024

https://github.com/erogol/fftnet

FFTNet vocoder implementation

deep-learning fftnet pytorch text2speech vocoder

Last synced: 06 Jan 2025

https://github.com/lyeoni/pretraining-for-language-understanding

Pre-training of Language Models for Language Understanding

language-model language-modeling language-understanding nlp pytorch pytorch-tutorial

Last synced: 06 Nov 2024

https://github.com/spcl/daceml

A Data-Centric Compiler for Machine Learning

compiler cuda deep-learning fpga high-performance-computing machine-learning pytorch

Last synced: 06 Nov 2024

https://github.com/sony/soundctm

Pytorch implementation of SoundCTM

audio-generation diffusion-models pytorch text-to-audio

Last synced: 17 Feb 2025

https://github.com/rose-stl-lab/torchts

Time series forecasting with PyTorch

deep-learning machine-learning pytorch pytorch-lightning time-series

Last synced: 15 Feb 2025