Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
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.
- GitHub: https://github.com/topics/pytorch
- Wikipedia: https://en.wikipedia.org/wiki/PyTorch
- Repo: https://github.com/pytorch/pytorch
- Created by: Facebook's AI Research lab (FAIR)
- Released: September 2016
- Related Topics: python, pytorch-tutorial,
- Last updated: 2024-11-15 00:24:09 UTC
- JSON Representation
https://github.com/microsoft/msrflute
Federated Learning Utilities and Tools for Experimentation
distributed-learning federated-learning gloo machine-learning nccl personalization privacy-tools pytorch simulation transformers-models
Last synced: 05 Nov 2024
https://github.com/atomicarchitects/equiformer
[ICLR'23 Spotlight] Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs
ai-for-science catalyst-design computational-chemistry deep-learning drug-discovery e3nn equivariant-graph-neural-network equivariant-neural-networks force-fields geometric-deep-learning graph-neural-networks interatomic-potentials machine-learning materials-science molecular-dynamics pytorch
Last synced: 13 Nov 2024
https://github.com/layumi/person-reid-triplet-loss
Person re-ID baseline with triplet loss
cnn image-retrieval open-reid personreid pytorch triplet-loss
Last synced: 31 Oct 2024
https://github.com/htdt/hyp_metric
Hyperbolic Vision Transformers: Combining Improvements in Metric Learning | Official repository
computer-vision machine-learning pytorch
Last synced: 15 Nov 2024
https://github.com/mblondel/fenchel-young-losses
Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses
loss-functions probabilistic-classification pytorch sklearn tensorflow
Last synced: 13 Oct 2024
https://github.com/obss/jury
Comprehensive NLP Evaluation System
datasets evaluate evaluation huggingface machine-learning metrics natural-language-processing nlp nlp-evaluation python pytorch transformers
Last synced: 12 Nov 2024
https://github.com/thnkim/OpenFacePytorch
PyTorch module to use OpenFace's nn4.small2.v1.t7 model
Last synced: 14 Nov 2024
https://github.com/divelab/good
GOOD: A Graph Out-of-Distribution Benchmark [NeurIPS 2022 Datasets and Benchmarks]
deep-learning distribution-shift graph-neural-networks graph-ood invariant-learning out-of-distribution-generalization pytorch pytorch-geometric
Last synced: 11 Nov 2024
https://github.com/suinleelab/path_explain
A repository for explaining feature attributions and feature interactions in deep neural networks.
explainable-ai interpretable-deep-learning machine-learning pytorch tensorflow
Last synced: 11 Nov 2024
https://github.com/willbrennan/semanticsegmentation
A framework for training segmentation models in pytorch on labelme annotations with pretrained examples of skin, cat, and pizza topping segmentation
birds bisenet bisenetv2 cats coco computer-vision labelme labelme-annotations pizza pizza-toppings pytorch segmentation semantic-segmentation skin-detection skin-segmentation torchvision
Last synced: 17 Nov 2024
https://github.com/belskikh/kekas
Just another DL library
deep-learning kek neural-networks pytorch
Last synced: 15 Nov 2024
https://github.com/ben-louis/deep-image-analogy-pytorch
A python implementation of Deep-Image-Analogy based on pytorch.
Last synced: 15 Nov 2024
https://github.com/thinklab-sjtu/eda-ai
Implementation of NeurIPS 2021 paper "On Joint Learning for Solving Placement and Routing in Chip Design" & NeurIPS 2022 paper "The Policy-gradient Placement and Generative Routing Neural Networks for Chip Design".
Last synced: 13 Nov 2024
https://github.com/drsleep/refinenet-pytorch
RefineNet-101 VOC in PyTorch
pytorch refinenet semantic-segmentation
Last synced: 16 Nov 2024
https://github.com/eriklindernoren/action-recognition
Exploration of different solutions to action recognition in video, using neural networks implemented in PyTorch.
action-recognition pytorch video-classification
Last synced: 11 Nov 2024
https://github.com/divelab/GOOD
GOOD: A Graph Out-of-Distribution Benchmark [NeurIPS 2022 Datasets and Benchmarks]
deep-learning distribution-shift graph-neural-networks graph-ood invariant-learning out-of-distribution-generalization pytorch pytorch-geometric
Last synced: 05 Nov 2024
https://github.com/mehmetgoren/feniks
A computer based NVR (Network Video Recorder) with AI capabilities
ai camera cctv face-recognition flv h265-hevc hls license-plate-recognition motion-detection nvr object-detection pytorch recoding rtmp rtsp streaming tensorflow video vms websocket
Last synced: 06 Nov 2024
https://github.com/shrubb/latent-pose-reenactment
The authors' implementation of the "Neural Head Reenactment with Latent Pose Descriptors" (CVPR 2020) paper.
avatar deep-learning face-reenactment facial-landmarks generative-model head-avatar head-reenactment landmark-detection pose-estimation pytorch self-supervised-learning talking-head voxceleb voxceleb2
Last synced: 29 Oct 2024
https://github.com/balavenkatesh3322/audio-pretrained-model
A collection of Audio and Speech pre-trained models.
audio audio-processing caffe keras keras-models keras-tensorflow machine-learning mxnet neural-network pre-trained pre-trained-model pre-training python3 pytorch pytorch-models speech-recognition speech-to-text tensorflow tensorflow-models
Last synced: 13 Oct 2024
https://github.com/nikhilbarhate99/min-decision-transformer
Minimal implementation of Decision Transformer: Reinforcement Learning via Sequence Modeling in PyTorch for mujoco control tasks in OpenAI gym
deep-learning deep-reinforcement-learning machine-learning mujoco offline-reinforcement-learning openai-gym pytorch pytorch-transformers reinforcement-learning robotics transformer
Last synced: 13 Nov 2024
https://github.com/sony/sqvae
Pytorch implementation of stochastically quantized variational autoencoder (SQ-VAE)
bayesian deep-generative-model generative-model gumbel-softmax machine-learning pytorch vae variational-autoencoder vector-quantization vq-vae
Last synced: 16 Nov 2024
https://github.com/wimlds-trojmiasto/detect-waste
AI to Combat Environmental Pollution - detecting plastic waste in the environment to combat environmental pollution and promote circular economy (Deep Learning, PyTorch)
cnn deep-learning detr efficientdet fastrcnn litter maskrcnn neural-networks object-detection python pytorch trash waste-detection
Last synced: 14 Nov 2024
https://github.com/hemile/storchastic
Stochastic Automatic Differentiation library for PyTorch.
gradient-estimators pytorch reparameterization stochastic-computation-graphs stochastic-deep-learning
Last synced: 22 Oct 2024
https://github.com/ChrisWaites/pyvacy
Differentially Private Optimization for PyTorch 👁🙅♀️
Last synced: 02 Nov 2024
https://github.com/snap-research/CAT
[CVPR 2021] Teachers Do More Than Teach: Compressing Image-to-Image Models (CAT)
compression cyclegan deep-learning gan gaugan image-to-image pix2pix pytorch
Last synced: 08 Aug 2024
https://github.com/snap-research/cat
[CVPR 2021] Teachers Do More Than Teach: Compressing Image-to-Image Models (CAT)
compression cyclegan deep-learning gan gaugan image-to-image pix2pix pytorch
Last synced: 15 Nov 2024
https://github.com/coqui-ai/Trainer
🐸 - A general purpose model trainer, as flexible as it gets
ai data-science deep-learning machine-learning pytorch
Last synced: 07 Aug 2024
https://github.com/akanametov/yolov8-face
YOLOv8 Face 🚀 in PyTorch > ONNX > CoreML > TFLite
computer-vision face-detection object-detection pytorch yolov8 yolov8-drone yolov8-face yolov8-football yolov8-parking
Last synced: 05 Aug 2024
https://github.com/fedml-ai/fedgraphnn
FedGraphNN: A Federated Learning Platform for Graph Neural Networks with MLOps Support. The previous research version is accepted to ICLR'2021 - DPML and MLSys'21 - GNNSys workshops.
deep-learning distributed-learning federated-learning federated-learning-framework fedml gnns graph-neural-networks machine-learning pytorch tensorflow
Last synced: 16 Nov 2024
https://github.com/tjuskyzhang/Scaled-YOLOv4-TensorRT
Got 100fps on TX2. Got 500fps on GeForce GTX 1660 Ti. If the project is useful to you, please Star it.
pytorch scaled-yolov4 scaled-yolov4-tenosrrt tensorrt tx2 yolo yolov4-csp yolov4-csp-tensorrt yolov4-large yolov4-large-tensorrt yolov4-tiny yolov4-tiny-tensorrt
Last synced: 09 Nov 2024
https://github.com/BMW-InnovationLab/BMW-Anonymization-API
This repository allows you to anonymize sensitive information in images/videos. The solution is fully compatible with the DL-based training/inference solutions that we already published/will publish for Object Detection and Semantic Segmentation.
anonymization-api anonymization-technique bmw computer-vision data-anonymization deep-learning docker image-transformation inference no-code object-detection openvino privacy-enhancing-technologies privacy-protection pytorch semantic-segmentation tensorflow tensorflow-training video video-anonymization
Last synced: 05 Nov 2024
https://github.com/maxent-ai/converse
Conversational text Analysis using various NLP techniques
callcenter-analysis conversational-ai emotion-recognition huggingface machine-learning nlp nlu pytorch scikit-learn sentiment-analysis spacy speech-to-text text text-mining topic-modeling transformers
Last synced: 01 Nov 2024
https://github.com/csinva/gan-vae-pretrained-pytorch
Pretrained GANs + VAEs + classifiers for MNIST/CIFAR in pytorch.
ai cifar cnn convolutional-neural-networks dcgan deep-learning gan gans generative-adversarial-network generative-adversarial-networks machine-learning ml mnist neural-network pretrained pretrained-models python pytorch pytorch-implementation statistics
Last synced: 15 Nov 2024
https://github.com/s3nh/text-detector
Tool which allow you to detect and translate text.
craft crnn deep-learning nlp ocr-recognition pytorch recognition scene-text-detection scene-text-detectors text text-processing text-recognition
Last synced: 03 Nov 2024
https://github.com/BMW-InnovationLab/BMW-Anonymization-API?referral=top-free-anonymization-tools-apis-and-open-source-models
This repository allows you to anonymize sensitive information in images/videos. The solution is fully compatible with the DL-based training/inference solutions that we already published/will publish for Object Detection and Semantic Segmentation.
anonymization-api anonymization-technique bmw computer-vision data-anonymization deep-learning docker image-transformation inference no-code object-detection openvino privacy-enhancing-technologies privacy-protection pytorch semantic-segmentation tensorflow tensorflow-training video video-anonymization
Last synced: 12 Aug 2024
https://github.com/david-knigge/ccnn
Code repository of the paper "Modelling Long Range Dependencies in ND: From Task-Specific to a General Purpose CNN" https://arxiv.org/abs/2301.10540.
convolutional-neural-networks deep-learning pytorch
Last synced: 28 Oct 2024
https://github.com/HennyJie/BrainGB
Officially Accepted to IEEE Transactions on Medical Imaging (TMI, IF: 11.037) - Special Issue on Geometric Deep Learning in Medical Imaging.
ai4health brain-networks gcn graph-convolutional-networks graph-neural-networks network-embedding pytorch
Last synced: 12 Nov 2024
https://github.com/riannevdberg/sylvester-flows
deep-learning normalizing-flows pytorch vae
Last synced: 05 Aug 2024
https://github.com/Ai-trainee/Traffic-Sign-Recognition-PyQt5-YOLOv5-GUI
This YOLOv5🚀😊 GUI road sign system uses MySQL💽, PyQt5🎨, PyTorch, CSS🌈. It has modules for login🔑, YOLOv5 setup📋, sign recognition🔍, database💾, and image processing🖼️. It supports diverse inputs, model switching, and enhancements like mosaic and mixup📈.
css mysql-database pyqt5 pytorch tensorrt-engine yolov5 yolov5-gui
Last synced: 09 Nov 2024
https://github.com/zubair-irshad/shapo
Pytorch code for ECCV'22 paper. ShAPO: Implicit Representations for Multi-Object Shape, Appearance and Pose Optimization
3d-computer-vision 3d-detection 3d-vision 6dof-pose computer-vision deep-learning implicit-neural-representation implicit-representions neural-fields optimization pose-estimation pytorch robotics scene-understanding shape-reconstruction signed-distance-functions texture-synthesis
Last synced: 16 Nov 2024
https://github.com/yerfor/syntaspeech
SyntaSpeech: Syntax-aware Generative Adversarial Text-to-Speech; IJCAI 2022; Official code
gan pytorch speech-synthesis tts
Last synced: 06 Nov 2024
https://github.com/dansuh17/alexnet-pytorch
Pytorch Implementation of AlexNet
Last synced: 14 Nov 2024
https://github.com/shi-labs/fcf-inpainting
[WACV 2023] Keys to Better Image Inpainting: Structure and Texture Go Hand in Hand
fast-fourier-convolution fcfgan image-inpainting pytorch stylegan2
Last synced: 15 Nov 2024
https://github.com/tomrunia/pytorchconv3d
I3D and 3D-ResNets in PyTorch
3d-models computer-vision deep-learning i3d pytorch resnet
Last synced: 13 Nov 2024
https://github.com/VSainteuf/pytorch-psetae
PyTorch implementation of the model presented in "Satellite Image Time Series Classification with Pixel-Set Encoders and Temporal Self-Attention"
agriculture computer-vision cvpr cvpr2020 deep-learning earth-observation machine-learning pytorch remote-sensing satellite-image self-attention spatio-temporal time-series-classification transformer-architecture
Last synced: 15 Nov 2024
https://github.com/pqhieu/jsis3d
[CVPR'19] JSIS3D: Joint Semantic-Instance Segmentation of 3D Point Clouds
cvpr deep-learning instance-segmentation point-cloud pytorch
Last synced: 28 Oct 2024
https://github.com/eriklindernoren/Action-Recognition
Exploration of different solutions to action recognition in video, using neural networks implemented in PyTorch.
action-recognition pytorch video-classification
Last synced: 06 Nov 2024
https://github.com/csyanbin/TPN-pytorch
Pytorch Code for ICLR19 paper: Learning to Propagate Labels: Transductive Propagation Network for Few-shot Learning.
few-shot-learning label-propagation meta-learning pytorch transductive-few-shot-learning
Last synced: 26 Oct 2024
https://github.com/d-li14/psconv
[ECCV 2020] PSConv: Squeezing Feature Pyramid into One Compact Poly-Scale Convolutional Layer
convolution eccv2020 feature-pyramids instance-segmentation mmdetection multi-scale object-detection pytorch
Last synced: 07 Nov 2024
https://github.com/mantasu/cs231n
Shortest solutions for CS231n 2021-2023
computer-vision convolutional-neural-networks cs231n deep-learning pytorch tensorflow visual-recognition
Last synced: 16 Nov 2024
https://github.com/acecoooool/dss-pytorch
:star: PyTorch implement of Deeply Supervised Salient Object Detection with Short Connection
python pytorch salient-object-detection
Last synced: 09 Nov 2024
https://github.com/d-li14/PSConv
[ECCV 2020] PSConv: Squeezing Feature Pyramid into One Compact Poly-Scale Convolutional Layer
convolution eccv2020 feature-pyramids instance-segmentation mmdetection multi-scale object-detection pytorch
Last synced: 14 Nov 2024
https://github.com/czczup/urst
[AAAI 2022] Towards Ultra-Resolution Neural Style Transfer via Thumbnail Instance Normalization
Last synced: 15 Nov 2024
https://github.com/xiaoyu258/GeoProj
Blind Geometric Distortion Correction on Images Through Deep Learning
deep-learning distortion-correction pytorch
Last synced: 25 Oct 2024
https://github.com/tugstugi/pytorch-dc-tts
Text to Speech with PyTorch (English and Mongolian)
convolutional-neural-networks deep-learning mongolian python pytorch speech-synthesis text-to-speech tts
Last synced: 15 Nov 2024
https://github.com/HEmile/storchastic
Stochastic Automatic Differentiation library for PyTorch.
gradient-estimators pytorch reparameterization stochastic-computation-graphs stochastic-deep-learning
Last synced: 05 Aug 2024
https://github.com/jzlianglu/pykaldi2
Yet another speech toolkit based on Kaldi and PyTorch
horovod kaldi pykaldi pytorch speech-toolkit
Last synced: 17 Aug 2024
https://github.com/phamquiluan/PubLayNet
ICDAR 2019: MaskRCNN on PubLayNet datasets. Paragraph detection, table detection, figure detection,...
document-layout-analysis figure-detection mask-rcnn object-detection paragraph-detection pretrained-models publaynet pytorch table-detection
Last synced: 08 Aug 2024
https://github.com/dengbocong/text-similarity
文本相似度(匹配)计算,提供Baseline、训练、推理、指标分析...代码包含TensorFlow/Pytorch双版本
bert deep-learning mechine-learing model nlp pytorch similarity text-classification transformer
Last synced: 15 Nov 2024
https://github.com/rayhane-mamah/efficient-vdvae
Official Pytorch and JAX implementation of "Efficient-VDVAE: Less is more"
image-generation jax paper python pytorch representation-learning unsupervised-learning vae
Last synced: 06 Nov 2024
https://github.com/j3soon/tbparse
Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow
machine-learning parser plot python pytorch reader tensorboard tensorboardx tensorflow
Last synced: 15 Nov 2024
https://github.com/markovmodel/deeptime
Deep learning meets molecular dynamics.
autoencoder computational-biology computational-chemistry data-analysis deep-learning dimension-reduction machine-learning markov-model python pytorch tensorflow time-series
Last synced: 18 Nov 2024
https://github.com/matajoh/fourier_feature_nets
Supplemental learning materials for "Fourier Feature Networks and Neural Volume Rendering"
computer-graphics computer-vision deep-learning pytorch
Last synced: 15 Nov 2024
https://github.com/guanyingc/sdps-net
Learning Based Uncalibrated Photometric Stereo for Non-Lambertian Surface (CVPR 2019 Oral)
deep-learning non-lambertian photometric-stereo pytorch uncalibrated
Last synced: 15 Nov 2024
https://github.com/kangyeolk/Paint-by-Sketch
Stable Diffusion-based image manipulation method with a sketch and reference image
computer-vision diffusion-models image-editing image-generation image-manipulation pytorch stable-diffusion
Last synced: 31 Oct 2024
https://github.com/kennardwang/vtuber-momosehiyori
😀 Live2D VTuber, made by Unity and Cubism. This project is based on TCP socket, OpenCV and deep learning
csharp cubism deep-learning dlib live2d opencv pycharm python3 pytorch tcp-socket unity vtuber
Last synced: 09 Oct 2024
https://github.com/Pilhyeon/BaSNet-pytorch
Official Pytorch Implementation of 'Background Suppression Network for Weakly-supervised Temporal Action Localization' (AAAI-20 Spotlight)
background-modeling deep-learning pytorch temporal-action-localization weakly-supervised-learning
Last synced: 13 Nov 2024
https://github.com/schroedinger-hat/imagegonord-pip
A tool that can convert your rgb images to nordtheme palette
color-palette color-scheme conversion-utility gruvbox image-processing machine-learning nordtheme-palette pillow pytorch recolor rgb-images
Last synced: 13 Nov 2024
https://github.com/Js-Mim/mss_pytorch
Singing Voice Separation via Recurrent Inference and Skip-Filtering Connections - PyTorch Implementation. Demo:
deep-learning mss-pytorch music-source-separation pytorch recurrent-neural-networks skip-filtering-connections
Last synced: 14 Nov 2024
https://github.com/atomicarchitects/equiformer_v2
[ICLR'24] EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representations
ai-for-science catalyst-design computational-chemistry computational-physics deep-learning drug-discovery e3nn equivariant-graph-neural-network equivariant-neural-networks force-fields geometric-deep-learning graph-neural-networks interatomic-potentials machine-learning materials-science molecular-dynamics pytorch
Last synced: 30 Oct 2024
https://github.com/ai-trainee/traffic-sign-recognition-pyqt5-yolov5-gui
This YOLOv5🚀😊 GUI road sign system uses MySQL💽, PyQt5🎨, PyTorch, CSS🌈. It has modules for login🔑, YOLOv5 setup📋, sign recognition🔍, database💾, and image processing🖼️. It supports diverse inputs, model switching, and enhancements like mosaic and mixup📈.
css mysql-database pyqt5 pytorch tensorrt-engine yolov5 yolov5-gui
Last synced: 12 Oct 2024
https://github.com/brendanhasz/probflow
A Python package for building Bayesian models with TensorFlow or PyTorch
bayesian bayesian-inference bayesian-methods bayesian-neural-networks bayesian-statistics data-science machine-learning python pytorch statistics tensorflow
Last synced: 13 Oct 2024
https://github.com/pyg-team/pyg-lib
Low-Level Graph Neural Network Operators for PyG
deep-learning geometric-deep-learning graph-convolutional-networks graph-neural-networks pytorch
Last synced: 12 Nov 2024
https://github.com/adobe-research/sam_inversion
[CVPR 2022] GAN inversion and editing with spatially-adaptive multiple latent layers
computer-graphics computer-vision deep-learning gan generative-adversarial-network image-manipulation machine-learning pytorch
Last synced: 14 Nov 2024
https://github.com/mithril-security/bastionlab
A simple framework for privacy-friendly data science collaboration
deep-learning eda multi-party privacy pytorch secure-enclave
Last synced: 09 Nov 2024
https://github.com/kakaobrain/autoclint
A specially designed light version of Fast AutoAugment
1st-place autocv-challenge autocv2-challenge autodl-challenge automated-machine-learning automl cnn computer-vision convolutional-neural-networks deep-learning fast-auto-augment image-classification pytorch
Last synced: 10 Nov 2024
https://github.com/plantnet/PlantNet-300K
[NeurIPS2021] A plant image dataset with high label ambiguity and a long-tailed distribution
dataset deep-learning plants pytorch
Last synced: 15 Nov 2024
https://github.com/ahirner/pytorch-retraining
Transfer Learning Shootout for PyTorch's model zoo (torchvision)
benchmark pytorch transfer-learning
Last synced: 15 Nov 2024
https://github.com/cure-lab/deciwatch
[ECCV 2022] Official implementation of the paper "DeciWatch: A Simple Baseline for 10x Efficient 2D and 3D Pose Estimation"
2d-human-pose 3d-body-recovery 3d-pose-estimation body-reconstruction deep-learning eccv eccv2022 efficiency efficient-inference efficient-neural-networks human-pose-estimation pose-estimation pytorch
Last synced: 14 Nov 2024
https://github.com/lukas-blecher/AutoMask
Automatic Rotoscoping/Masking tool powered with Machine Learning for Blender (WIP)
blender blender-addon blender-python deep-learning machine-learning mask-segment masking motion-tracking pytorch rotoscoping
Last synced: 16 Nov 2024
https://github.com/balavenkatesh3322/nlp-pretrained-model
A collection of Natural language processing pre-trained models.
deep-learning deep-neural-networks keras machine-learning model mxnet natural-language-processing neural-networks nlp nlp-machine-learning python python3 pytorch tensorflow text text-classification text-to-number text-to-speech
Last synced: 10 Nov 2024
https://github.com/nadeemlab/DeepLIIF
Deep Learning Inferred Multiplex ImmunoFluorescence for IHC Image Quantification (https://deepliif.org) [Nature Machine Intelligence'22, CVPR'22, MICCAI'23, Histopathology'23, MICCAI'24]
ai-ready cell-biology cell-segmentation cloud-native-applications datasets deep-learning digital-pathology image-segmentation immunohistochemical-images immunohistochemistry multiplex multitask-learning pathology pathology-gan pathology-image pytorch segmentation stitching super-resolution whole-slide-imaging
Last synced: 13 Nov 2024
https://github.com/IlyaGusev/summarus
Models for automatic abstractive summarization
deep-learning machine-learning nlp pytorch summarization
Last synced: 04 Nov 2024
https://github.com/daniil-osokin/gccpm-look-into-person-cvpr19.pytorch
Fast and accurate single-person pose estimation, ranked 10th at CVPR'19 LIP challenge. Contains implementation of "Global Context for Convolutional Pose Machines" paper.
convolutional-pose-machines cvpr2019 deep-learning human-pose human-pose-estimation lightweight look-into-person openvino pose-estimation pytorch real-time single-person
Last synced: 31 Oct 2024
https://github.com/spytensor/detect_steel_bar
CCFDF AI 数钢筋大赛
detection gangjin pytorch retinanet
Last synced: 06 Nov 2024
https://github.com/chrischoy/makepytorchplusplus
How and why you want to make your pytorch CUDA/CPP extension with a Makefile
cpp-extension pytorch pytorch-cpp tutorial
Last synced: 07 Nov 2024
https://github.com/chaitjo/learning-tsp
Code for the paper 'Learning TSP Requires Rethinking Generalization' (CP 2021)
combinatorial-optimization deep-learning graph-neural-networks graph-representation-learning pytorch transformers travelling-salesman-problem tsp
Last synced: 16 Nov 2024
https://github.com/liuzuxin/osrl
🤖 Elegant implementations of offline safe RL algorithms in PyTorch
bc-safe cdt cpq library offline-rl offline-safe-rl pytorch reinforcement-learning robotics safe-rl
Last synced: 12 Nov 2024
https://github.com/idsia/modern-srwm
Official repository for the paper "A Modern Self-Referential Weight Matrix That Learns to Modify Itself" (ICML 2022 & NeurIPS 2021 Deep RL Workshop) and "Accelerating Neural Self-Improvement via Bootstrapping" (ICLR 2023 Workshop)
fast-weight-programmers fast-weights pytorch reinforcement-learning rnn-based-few-shot-learning self-referential-learning self-referential-weight-matrix torchbeast transformers
Last synced: 11 Nov 2024
https://github.com/zudi-lin/pytorch_connectomics
PyTorch Connectomics: segmentation toolbox for EM connectomics
biomedical-image-processing computer-vision connectomics deep-learning microscopy neuroscience pytorch segmentation
Last synced: 12 Nov 2024
https://github.com/IDSIA/modern-srwm
Official repository for the paper "A Modern Self-Referential Weight Matrix That Learns to Modify Itself" (ICML 2022 & NeurIPS 2021 Deep RL Workshop) and "Accelerating Neural Self-Improvement via Bootstrapping" (ICLR 2023 Workshop)
fast-weight-programmers fast-weights pytorch reinforcement-learning rnn-based-few-shot-learning self-referential-learning self-referential-weight-matrix torchbeast transformers
Last synced: 10 Nov 2024
https://github.com/lukas-blecher/automask
Automatic Rotoscoping/Masking tool powered with Machine Learning for Blender (WIP)
blender blender-addon blender-python deep-learning machine-learning mask-segment masking motion-tracking pytorch rotoscoping
Last synced: 01 Nov 2024
https://github.com/ymcui/lert
LERT: A Linguistically-motivated Pre-trained Language Model(语言学信息增强的预训练模型LERT)
bert lert nlp plm pre-train pytorch tensorflow transformer
Last synced: 28 Oct 2024
https://github.com/plantnet/plantnet-300k
[NeurIPS2021] A plant image dataset with high label ambiguity and a long-tailed distribution
dataset deep-learning plants pytorch
Last synced: 14 Nov 2024
https://github.com/pxiangwu/MotionNet
CVPR 2020, "MotionNet: Joint Perception and Motion Prediction for Autonomous Driving Based on Bird's Eye View Maps"
autonomous-driving cvpr2020 motion-prediction perception pytorch
Last synced: 28 Oct 2024
https://github.com/balavenkatesh3322/NLP-pretrained-model
A collection of Natural language processing pre-trained models.
deep-learning deep-neural-networks keras machine-learning model mxnet natural-language-processing neural-networks nlp nlp-machine-learning python python3 pytorch tensorflow text text-classification text-to-number text-to-speech
Last synced: 07 Aug 2024
https://github.com/clovaai/rebias
Official Pytorch implementation of ReBias (Learning De-biased Representations with Biased Representations), ICML 2020
bias-generalisation bias-generalization deep-learning icml icml-2020 icml2020 machine-learning pytorch rebias
Last synced: 12 Nov 2024
https://github.com/sunset1995/pytorch-layoutnet
Pytorch implementation of LayoutNet.
computer-vision layoutnet pytorch pytorch-layoutnet room-layout
Last synced: 11 Nov 2024