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/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/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/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/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/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/young-geng/cql

Conservative Q Learning on top of SAC

pytorch reinforcement-learning

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

(Personalized) Page-Rank computation using PyTorch

gpu pagerank personalized-pagerank pytorch

Last synced: 18 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/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/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/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/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/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/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/ai4iacts/hexagdly

Process hexagonally sampled data with PyTorch

deep-learning hexagonal-grids pytorch

Last synced: 01 Nov 2024

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

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

deeplearning education machine-learning pytorch textbook

Last synced: 05 Nov 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/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/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/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/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/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/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/yuanzhoulvpi2017/quick_sentence_transformers

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

bert pytorch sentence-transformers transformer transformers

Last synced: 08 Nov 2024

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/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/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/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/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/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/borhanmorphy/fastface

Light Face Detection using PyTorch Lightning

edge-ai facedetection pytorch pytorch-lightning

Last synced: 21 Jan 2025

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/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/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/lucasbotang/gradnorm

PyTorch implementation of the GradNorm

deep-learning gradnorm multitask-learning python pytorch

Last synced: 17 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/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/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/intel/torch-ccl

oneCCL Bindings for Pytorch*

oneccl pytorch

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

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

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

deep-learning keras pytorch tensorflow2

Last synced: 07 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/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/sony/soundctm

Pytorch implementation of SoundCTM

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

Last synced: 17 Feb 2025

https://github.com/erogol/fftnet

FFTNet vocoder implementation

deep-learning fftnet pytorch text2speech vocoder

Last synced: 06 Jan 2025

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