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-10 00:23:23 UTC
- JSON Representation
https://github.com/yeyupiaoling/voiceprintrecognition-pytorch
This project uses a variety of advanced voiceprint recognition models such as EcapaTdnn, ResNetSE, ERes2Net, CAM++, etc. It is not excluded that more models will be supported in the future. At the same time, this project also supports MelSpectrogram, Spectrogram data preprocessing methods
arcface ecapa-tdnn pytorch speaker-recognition voice-recognition
Last synced: 31 Oct 2024
https://github.com/simonvandenhende/multi-task-learning-pytorch
PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).
computer-vision eccv2020 multi-task-learning nyud pascal pytorch scene-understanding segmentation
Last synced: 14 Nov 2024
https://github.com/kwea123/VTuber_Unity
Use Unity 3D character and Python deep learning algorithms to stream as a VTuber!
dlib face-detection face-landmark-detection face-pose gaze-estimation python pytorch unity unity-chan vrm vtuber
Last synced: 10 Nov 2024
https://github.com/oaid/autokernel
AutoKernel 是一个简单易用，低门槛的自动算子优化工具，提高深度学习算法部署效率。
auto autosearch deep-learning halide optimization pytorch reinforcement-learning tengine tensor tensorflow tvm
Last synced: 05 Nov 2024
https://github.com/pytorch/benchmark
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
Last synced: 07 Oct 2024
https://github.com/youngwanLEE/centermask2
[CVPR 2020] CenterMask : Real-time Anchor-Free Instance Segmentation
anchor-free centermask cvpr2020 detectron2 instance-segmentation object-detection pytorch real-time vovnet vovnetv2
Last synced: 26 Oct 2024
https://github.com/MediaBrain-SJTU/MING
明医 (MING)：中文医疗问诊大模型
consultation huggingface llm medical pytorch transformers
Last synced: 03 Aug 2024
https://github.com/parrt/tensor-sensor
The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.
debugging deep-learning jax matrix numpy python pytorch tensorflow tracing vector
Last synced: 15 Oct 2024
https://github.com/jeya-maria-jose/Medical-Transformer
Official Pytorch Code for "Medical Transformer: Gated Axial-Attention for Medical Image Segmentation" - MICCAI 2021
attention deep-learning medical-image-analysis medical-imaging pytorch segmentation transformer transformers
Last synced: 03 Aug 2024
https://github.com/vandit15/Class-balanced-loss-pytorch
Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"
computer-vision cvpr2019 deep-learning loss-functions pytorch
Last synced: 03 Aug 2024
https://github.com/tusen-ai/sst
Code for a series of work in LiDAR perception, including SST (CVPR 22), FSD (NeurIPS 22), FSD++ (TPAMI 23), FSDv2, and CTRL (ICCV 23, oral).
3d-object-detection autonomous-driving pytorch
Last synced: 08 Nov 2024
https://github.com/tusen-ai/SST
Code for a series of work in LiDAR perception, including SST (CVPR 22), FSD (NeurIPS 22), FSD++ (TPAMI 23), FSDv2, and CTRL (ICCV 23, oral).
3d-object-detection autonomous-driving pytorch
Last synced: 28 Oct 2024
https://github.com/higgsfield/capsule-network-tutorial
Pytorch easy-to-follow Capsule Network tutorial
capsule-network clean-code easy-to-use pytorch pytorch-tutorials
Last synced: 11 Nov 2024
https://github.com/microsoft/cameratraps
PyTorch Wildlife: a Collaborative Deep Learning Framework for Conservation.
camera-traps computer-vision conservation machine-learning megadetector pytorch pytorch-wildlife wildlife
Last synced: 13 Nov 2024
https://github.com/mit-han-lab/tinyengine
[NeurIPS 2020] MCUNet: Tiny Deep Learning on IoT Devices; [NeurIPS 2021] MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learning; [NeurIPS 2022] MCUNetV3: On-Device Training Under 256KB Memory
c codegenerator cpp deep-learning edge-computing microcontroller neural-architecture-search pytorch quantization tinyml
Last synced: 27 Oct 2024
https://github.com/Spijkervet/SimCLR
PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.
contrastive-learning pytorch representation-learning simclr unsupervised-learning
Last synced: 05 Nov 2024
https://github.com/kuan-wang/pytorch-mobilenet-v3
MobileNetV3 in pytorch and ImageNet pretrained models
classification imagenet mobilenet mobilenetv2 mobilenetv3 pytorch
Last synced: 31 Oct 2024
https://github.com/higgsfield/Capsule-Network-Tutorial
Pytorch easy-to-follow Capsule Network tutorial
capsule-network clean-code easy-to-use pytorch pytorch-tutorials
Last synced: 28 Oct 2024
https://github.com/kaituoxu/Speech-Transformer
A PyTorch implementation of Speech Transformer, an End-to-End ASR with Transformer network on Mandarin Chinese.
asr attention attention-is-all-you-need end-to-end pytorch self-attention transformer
Last synced: 07 Aug 2024
https://github.com/spijkervet/simclr
PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.
contrastive-learning pytorch representation-learning simclr unsupervised-learning
Last synced: 11 Oct 2024
https://github.com/shibing624/similarities
Similarities: a toolkit for similarity calculation and semantic search. 相似度计算、匹配搜索工具包，支持亿级数据文搜文、文搜图、图搜图，python3开发，开箱即用。
bm25 deep-learning faiss image-search image-similarity matching nlp pytorch search-engine similarity similarity-search text-matching
Last synced: 22 Oct 2024
https://github.com/xinntao/facexlib
FaceXlib aims at providing ready-to-use face-related functions based on current STOA open-source methods.
alignment assessment deep-learning detection face headpose matting parsing pytorch recognition tracking
Last synced: 22 Oct 2024
https://github.com/speedinghzl/pytorch-segmentation-toolbox
PyTorch Implementations for DeeplabV3 and PSPNet
cityscape-dataset deeplabv3 pspnet pytorch pytorch-segmentation-toolbox semantic-segmentation
Last synced: 29 Oct 2024
https://github.com/Denis2054/Transformers-for-NLP-2nd-Edition
Transformer models from BERT to GPT-4, environments from Hugging Face to OpenAI. Fine-tuning, training, and prompt engineering examples. A bonus section with ChatGPT, GPT-3.5-turbo, GPT-4, and DALL-E including jump starting GPT-4, speech-to-text, text-to-speech, text to image generation with DALL-E, Google Cloud AI,HuggingGPT, and more
bert chatgpt chatgpt-api dall-e dall-e-api deep-learning gpt-3-5-turbo gpt-4 gpt-4-api huggingface-transformers machine-learning natural-language-processing nlp openai python pytorch roberta-model transformers trax
Last synced: 24 Oct 2024
https://github.com/SimonVandenhende/Multi-Task-Learning-PyTorch
PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).
computer-vision eccv2020 multi-task-learning nyud pascal pytorch scene-understanding segmentation
Last synced: 13 Nov 2024
https://github.com/plemeri/transparent-background
This is a background removing tool powered by InSPyReNet (ACCV 2022)
background-removal deep-learning dichotomous-image-segmentation image-matting image-processing photo-editing python pytorch remove-background remove-background-image remove-background-video salient-object-detection video-editing
Last synced: 05 Nov 2024
https://github.com/berniwal/swin-transformer-pytorch
Implementation of the Swin Transformer in PyTorch.
artificial-intelligence attention-model deep-learning machine-learning pytorch transformer-architecture transformer-pytorch
Last synced: 07 Aug 2024
https://github.com/fastnlp/fasthan
fastHan是基于fastNLP与pytorch实现的中文自然语言处理工具，像spacy一样调用方便。
bert cws fastnlp joint-model ner parser pos python pytorch
Last synced: 05 Nov 2024
https://github.com/positive666/yolo_research
based on yolo-high-level project (detect\pose\classify\segment\):include yolov5\yolov7\yolov8\ core ,improvement research ,SwintransformV2 and Attention Series. training skills, business customization, engineering deployment C
attention-mechanism custom-networ deepstream object-detection pytorch swintransformer yolov5 yolov7 yolov8
Last synced: 09 Nov 2024
https://github.com/rusty1s/pytorch_cluster
PyTorch Extension Library of Optimized Graph Cluster Algorithms
cluster-algorithms geometric-deep-learning graph-neural-networks pytorch
Last synced: 22 Oct 2024
https://github.com/hengyuan-hu/bottom-up-attention-vqa
An efficient PyTorch implementation of the winning entry of the 2017 VQA Challenge.
bottom-up-attention pytorch vqa
Last synced: 07 Nov 2024
https://github.com/loicland/superpoint_graph
Large-scale Point Cloud Semantic Segmentation with Superpoint Graphs
clustering large-scale lidar partition ply-files point-cloud pytorch segmentation semantic semantic-segmentation superpoint-graphs
Last synced: 14 Nov 2024
https://github.com/teddykoker/torchsort
Fast, differentiable sorting and ranking in PyTorch
cuda-kernel pytorch ranking sort
Last synced: 03 Aug 2024
https://github.com/SongweiGe/rich-text-to-image
Rich-Text-to-Image Generation
computer-vision diffusion-models pytorch rich-text text-to-image-generation
Last synced: 31 Oct 2024
https://github.com/ikergarcia1996/self-driving-car-in-video-games
A deep neural network that learns to drive in video games
autonomous-driving deep-learning deep-neural-network machine-learning neural-network pretrained-models python pytorch self-driving-car supervised-learning video-games videogame videogame-bot
Last synced: 01 Nov 2024
https://github.com/benedekrozemberczki/simgnn
A PyTorch implementation of "SimGNN: A Neural Network Approach to Fast Graph Similarity Computation" (WSDM 2019).
attention-mechanism deep-learning gcn ged gnn graph-attention graph-classification graph-convolution graph-edit-distance graph-embedding graph-similarity machine-learning network-embedding neural-network pytorch simgnn sklearn tensor-network tensorflow wsdm
Last synced: 10 Oct 2024
https://github.com/tencent/patrickstar
PatrickStar enables Larger, Faster, Greener Pretrained Models for NLP and democratizes AI for everyone.
bert gpt nlp pretrained-models pytorch
Last synced: 13 Nov 2024
https://github.com/G-U-N/PyCIL
PyCIL: A Python Toolbox for Class-Incremental Learning
continual-learning deep-learning incremental-learning lifelong-learning machine-learning pytorch reproducible-research
Last synced: 03 Aug 2024
https://github.com/asyml/texar-pytorch
Integrating the Best of TF into PyTorch, for Machine Learning, Natural Language Processing, and Text Generation. This is part of the CASL project: http://casl-project.ai/
bert casl-project data-processing deep-learning dialog-systems gpt-2 machine-learning machine-translation natural-language-processing python pytorch roberta texar texar-pytorch text-data text-generation xlnet
Last synced: 12 Nov 2024
https://github.com/wmcnally/kapao
KAPAO is an efficient single-stage human pose estimation model that detects keypoints and poses as objects and fuses the detections to predict human poses.
deep-learning human-pose-estimation pose-estimation pytorch yolo
Last synced: 09 Nov 2024
https://github.com/fastnlp/fastHan
fastHan是基于fastNLP与pytorch实现的中文自然语言处理工具，像spacy一样调用方便。
bert cws fastnlp joint-model ner parser pos python pytorch
Last synced: 27 Oct 2024
https://github.com/fregu856/deeplabv3
PyTorch implementation of DeepLabV3, trained on the Cityscapes dataset.
autonomous-driving computer-vision deep-learning machine-learning pytorch semantic-segmentation
Last synced: 07 Aug 2024
https://github.com/sail-sg/Adan
Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
adan artificial-intelligence bert-model convnext cuda-programming deep-learning diffusion dreamfusion fairseq gpt2 llm-training llms mae moe optimizer pytorch resnet timm transformer-xl vit
Last synced: 05 Nov 2024
https://github.com/Tencent/PatrickStar
PatrickStar enables Larger, Faster, Greener Pretrained Models for NLP and democratizes AI for everyone.
bert gpt nlp pretrained-models pytorch
Last synced: 03 Aug 2024
https://github.com/guochengqian/PointNeXt
[NeurIPS'22] PointNeXt: Revisiting PointNet++ with Improved Training and Scaling Strategies
3d-vision pointcloud pointnet2 pytorch
Last synced: 28 Oct 2024
https://github.com/soapisnotfat/super-resolution
collection of super-resolution models & algorithms
fsrcnn pytorch resolution-models srcnn srgan sup-pixel-cnn
Last synced: 05 Aug 2024
https://github.com/msnh2012/Msnhnet
🔥 (yolov3 yolov4 yolov5 unet ...)A mini pytorch inference framework which inspired from darknet.
darknet inference-engine jetson-nx mobilenetv2 mobilenetyolo pytorch yolov3 yolov4 yolov5
Last synced: 09 Nov 2024
https://github.com/chenyuntc/pytorch-best-practice
A Guidance on PyTorch Coding Style Based on Kaggle Dogs vs. Cats
image-classification pytorch visdom
Last synced: 13 Nov 2024
https://github.com/pdebench/pdebench
PDEBench: An Extensive Benchmark for Scientific Machine Learning
ai autoregressive-models benchmark deep-learning fluid-dynamics jax machine-learning navier-stokes-equations neural-networks neural-operators partial-differential-equations physics-informed-neural-networks pytorch scientific scientific-computing sciml simulation
Last synced: 03 Nov 2024
https://github.com/williamfalcon/test-tube
Python library to easily log experiments and parallelize hyperparameter search for neural networks
caffe caffe2 chainer data-science deep-learning grid-search hyperparameter-optimization keras machine-learning neural-networks pytorch random-search tensorflow
Last synced: 12 Nov 2024
https://github.com/williamFalcon/test-tube
Python library to easily log experiments and parallelize hyperparameter search for neural networks
caffe caffe2 chainer data-science deep-learning grid-search hyperparameter-optimization keras machine-learning neural-networks pytorch random-search tensorflow
Last synced: 30 Oct 2024
https://github.com/jackroos/VL-BERT
Code for ICLR 2020 paper "VL-BERT: Pre-training of Generic Visual-Linguistic Representations".
bert iclr2020 pre-training pytorch representation-learning self-supervised-learning vision-and-language vl-bert
Last synced: 03 Nov 2024
https://github.com/fboulnois/stable-diffusion-docker
Run the official Stable Diffusion releases in a Docker container with txt2img, img2img, depth2img, pix2pix, upscale4x, and inpaint.
dall-e dalle diffusion docker generative-art huggingface image-generation inpainting midjourney pytorch stable-diffusion tensorflow text-to-image
Last synced: 12 Nov 2024
https://github.com/paddlepaddle/x2paddle
Deep learning model converter for PaddlePaddle. (『飞桨』深度学习模型转换工具)
caffe model-converter onnx paddlepaddle pytorch tensorflow x2paddle-model-zoo
Last synced: 13 Nov 2024
https://github.com/kymatio/kymatio
Wavelet scattering transforms in Python with GPU acceleration
deep-learning keras numpy pytorch scattering-transform scikit-learn tensorflow wavelets
Last synced: 10 Oct 2024
https://github.com/allentdan/libtorchtutorials
This is a code repository for pytorch c++ (or libtorch) tutorial.
classification cpp libtorch pytorch pytorch-cpp resnet segmentation tutorial-code unet vgg yolov4 yolov4-tiny
Last synced: 01 Nov 2024
https://github.com/PaddlePaddle/X2Paddle
Deep learning model converter for PaddlePaddle. (『飞桨』深度学习模型转换工具)
caffe model-converter onnx paddlepaddle pytorch tensorflow x2paddle-model-zoo
Last synced: 06 Nov 2024
https://github.com/chungyiweng/humannerf
HumanNeRF turns a monocular video of moving people into a 360 free-viewpoint video.
cvpr2022 humannerf nerf pytorch
Last synced: 07 Nov 2024
https://github.com/maximumentropy/seq2seq-pytorch
Sequence to Sequence Models with PyTorch
deep-learning pytorch rnn seq2seq
Last synced: 13 Nov 2024
https://github.com/nlp-uoregon/trankit
Trankit is a Light-Weight Transformer-based Python Toolkit for Multilingual Natural Language Processing
adapters artificial-intelligence deeplearning dependency-parsing language-model lemmatization machine-learning morphological-tagging multilingual natural-language-processing nlp part-of-speech-tagging pytorch sentence-segmentation tokenization universal-dependencies xlm-roberta
Last synced: 20 Oct 2024
https://github.com/quic/sense
Enhance your application with the ability to see and interact with humans using any RGB camera.
activity-recognition calorie-estimation computer-vision deep-learning fitness-app gesture-recognition neural-networks pytorch video
Last synced: 05 Aug 2024
https://github.com/ikergarcia1996/Self-Driving-Car-in-Video-Games
A deep neural network that learns to drive in video games
autonomous-driving deep-learning deep-neural-network machine-learning neural-network pretrained-models python pytorch self-driving-car supervised-learning video-games videogame videogame-bot
Last synced: 24 Aug 2024
https://github.com/drsleep/light-weight-refinenet
Light-Weight RefineNet for Real-Time Semantic Segmentation
deep-learning light-weight pytorch refinenet semantic-segmentation
Last synced: 09 Nov 2024
https://github.com/MisaOgura/flashtorch
Visualization toolkit for neural networks in PyTorch! Demo -->
cnn deep-learning explainability interpretability machine-learning neural-networks pytorch visualization
Last synced: 30 Oct 2024
https://github.com/DrSleep/light-weight-refinenet
Light-Weight RefineNet for Real-Time Semantic Segmentation
deep-learning light-weight pytorch refinenet semantic-segmentation
Last synced: 04 Aug 2024
https://github.com/SMILELab-FL/FedLab
A flexible Federated Learning Framework based on PyTorch, simplifying your Federated Learning research.
deep-learning federated-learning federated-learning-framework fedlab machine-learning pytorch pytorch-federated-learning
Last synced: 12 Nov 2024
https://github.com/ritchieng/deep-learning-wizard
Open source guides/codes for mastering deep learning to deploying deep learning in production in PyTorch, Python, Apptainer, and more.
apptainer deep-learning llamaindex ollama python3 pytorch
Last synced: 10 Oct 2024
https://github.com/primeqa/primeqa
The prime repository for state-of-the-art Multilingual Question Answering research and development.
ai bert dpr ibm ibm-research-ai language-model machine-learning natural-language-processing neural-information-retrieval neural-search nlp python pytorch question-answering semantic-search squad transfer-learning
Last synced: 22 Oct 2024
https://github.com/creafz/pytorch-cnn-finetune
Fine-tune pretrained Convolutional Neural Networks with PyTorch
convolutional-neural-networks deep-learning pretrained-models pytorch
Last synced: 13 Nov 2024
https://github.com/microsoft/Tutel
Tutel MoE: An Optimized Mixture-of-Experts Implementation
mixture-of-experts moe nlp pytorch transformer
Last synced: 01 Nov 2024
https://github.com/modal-labs/modal-examples
Examples of programs built using Modal
cloud distributed gpu machine-learning modal python pytorch serverless stable-diffusion web
Last synced: 05 Nov 2024
https://github.com/hkchengrex/cutie
[CVPR 2024 Highlight] Putting the Object Back Into Video Object Segmentation
computer-vision cvpr2024 deep-learning pytorch segmentation video-editing video-object-segmentation video-segmentation
Last synced: 10 Nov 2024
https://github.com/hkchengrex/Cutie
[CVPR 2024 Highlight] Putting the Object Back Into Video Object Segmentation
computer-vision cvpr2024 deep-learning pytorch segmentation video-editing video-object-segmentation video-segmentation
Last synced: 08 Nov 2024
https://github.com/mhamilton723/stego
Unsupervised Semantic Segmentation by Distilling Feature Correspondences
computer-vision deep-learning iclr2022 pytorch semantic-segmentation unsupervised-learning
Last synced: 01 Nov 2024
https://github.com/zhunzhong07/Random-Erasing
Random Erasing Data Augmentation. Experiments on CIFAR10, CIFAR100 and Fashion-MNIST
aaai2020 data-augmentation image-classification object-detection person-re-identification pytorch
Last synced: 12 Nov 2024
https://github.com/cadene/vqa.pytorch
Visual Question Answering in Pytorch
clevr coco deep-learning pytorch resnet skipthoughts torch vgenome vqa
Last synced: 14 Nov 2024
https://github.com/linksense/LightNet
LightNet: Light-weight Networks for Semantic Image Segmentation (Cityscapes and Mapillary Vistas Dataset)
cityscapes deeplabv3 deeplabv3plus inplace-activated-batchnorm light-weight-net light-weight-networks mapillary-vistas-dataset mobilenet mobilenet-v2 mobilenetv2 mobilenetv2plus pytorch rfmobilenetv2plus scse-aspp segmentation semantic-context-loss semantic-segmentation shufflenet wide-residual-networks
Last synced: 05 Aug 2024
https://github.com/microsoft/tutel
Tutel MoE: An Optimized Mixture-of-Experts Implementation
mixture-of-experts moe nlp pytorch transformer
Last synced: 07 Oct 2024
https://github.com/microsoft/CameraTraps
PyTorch Wildlife: a Collaborative Deep Learning Framework for Conservation.
camera-traps computer-vision conservation machine-learning megadetector pytorch pytorch-wildlife wildlife
Last synced: 08 Aug 2024
https://github.com/weiaicunzai/Bag_of_Tricks_for_Image_Classification_with_Convolutional_Neural_Networks
experiments on Paper <Bag of Tricks for Image Classification with Convolutional Neural Networks> and other useful tricks to improve CNN acc
Last synced: 04 Nov 2024
https://github.com/Cadene/vqa.pytorch
Visual Question Answering in Pytorch
clevr coco deep-learning pytorch resnet skipthoughts torch vgenome vqa
Last synced: 02 Nov 2024
https://github.com/jia-zhuang/pytorch-multi-gpu-training
整理 pytorch 单机多 GPU 训练方法与原理
Last synced: 03 Nov 2024
https://github.com/lonepatient/albert_pytorch
A Lite Bert For Self-Supervised Learning Language Representations
albert bert language-model mask ngram nlp pytorch
Last synced: 13 Nov 2024
https://github.com/benedekrozemberczki/SimGNN
A PyTorch implementation of "SimGNN: A Neural Network Approach to Fast Graph Similarity Computation" (WSDM 2019).
attention-mechanism deep-learning gcn ged gnn graph-attention graph-classification graph-convolution graph-edit-distance graph-embedding graph-similarity machine-learning network-embedding neural-network pytorch simgnn sklearn tensor-network tensorflow wsdm
Last synced: 08 Aug 2024
https://github.com/run-house/runhouse
The fastest way to iterate and deploy AI workloads on your own infra. Unobtrusive, debuggable, PyTorch-like APIs.
api artificial-intelligence aws azure collaboration data-science deployment distributed fastapi gcp infrastructure machine-learning middleware observability python pytorch ray sagemaker serverless
Last synced: 11 Oct 2024
https://github.com/lonePatient/albert_pytorch
A Lite Bert For Self-Supervised Learning Language Representations
albert bert language-model mask ngram nlp pytorch
Last synced: 02 Nov 2024
https://github.com/onepanelio/onepanel
The open source, end-to-end computer vision platform. Label, build, train, tune, deploy and automate in a unified platform that runs on any cloud and on-premises.
ai aiops annotation computer-vision deeplearning etl hyperparameter-tuning inference jupyterlab labeling machinelearning mlops pipelines pytorch tensorboard tensorflow training workflows
Last synced: 05 Nov 2024
https://github.com/zhiqwang/yolort
yolort is a runtime stack for yolov5 on specialized accelerators such as tensorrt, libtorch, onnxruntime, tvm and ncnn.
deployment detection graghsurgeon inference jit libtorch lightning ncnn nms onnx onnxruntime pytorch tensorrt torchscript trt tvm visualization yolo yolort yolov5
Last synced: 28 Oct 2024
https://github.com/ethanhe42/kl-loss
Bounding Box Regression with Uncertainty for Accurate Object Detection (CVPR'19)
detection detection-algorithm detection-model detection-network object-detection pytorch
Last synced: 01 Nov 2024
https://github.com/wohlert/semi-supervised-pytorch
Implementations of various VAE-based semi-supervised and generative models in PyTorch
generative-models pytorch semi-supervised-learning
Last synced: 14 Nov 2024
https://github.com/mosaicml/streaming
A Data Streaming Library for Efficient Neural Network Training
dataset deep-learning machine-learning neural-network pytorch streaming
Last synced: 11 Nov 2024
https://github.com/yl-1993/learn-to-cluster
Learning to Cluster Faces (CVPR 2019, CVPR 2020)
cvpr2019 cvpr2020 face-clustering face-recognition gcn pytorch
Last synced: 02 Nov 2024
https://github.com/fidler-lab/polyrnn-pp-pytorch
PyTorch training/tool code for Polygon-RNN++ (CVPR 2018)
annotation cvpr2018 deep-learning instance-segmentation labelling polygon-rnn pytorch
Last synced: 07 Aug 2024
https://github.com/cresset-template/cresset
Template repository to build PyTorch projects from source on any version of PyTorch/CUDA/cuDNN.
build cuda deep-learning deep-learning-tutorial docker docker-compose machine-learning makefile mlops mlops-template python pytorch source source-python template template-repository wheel
Last synced: 05 Nov 2024
https://github.com/bobbens/sketch_simplification
Models and code related to sketch simplification of rough sketches.
convolutional-neural-networks deep-learning pytorch siggraph sketch torch
Last synced: 09 Nov 2024
https://github.com/stevenygd/PointFlow
PointFlow : 3D Point Cloud Generation with Continuous Normalizing Flows
3d-point-clouds computer-vision continuous-normalizing-flows machine-learning pytorch shapes
Last synced: 05 Aug 2024
https://github.com/AstraZeneca/chemicalx
A PyTorch and TorchDrug based deep learning library for drug pair scoring. (KDD 2022)
biology chemistry deep-chemistry deep-learning drug drug-discovery drug-interaction drug-pair geometric-deep-learning geometry graph-neural-network machine-learning pharma polypharmacy pytorch smiles smiles-strings torch torchdrug
Last synced: 30 Oct 2024
https://github.com/alfredxiangwu/lightcnn
A Light CNN for Deep Face Representation with Noisy Labels, TIFS 2018
face-recognition lightcnn pytorch
Last synced: 09 Nov 2024
