dusty-nv / pytorch-classificationLinks
Training of image classification models with PyTorch
☆32Updated last year
Alternatives and similar repositories for pytorch-classification
Users that are interested in pytorch-classification are comparing it to the libraries listed below
Sorting:
- Training of semantic segmentation networks with PyTorch☆35Updated 4 years ago
- Training of object detection networks with PyTorch☆16Updated last year
- GUI tool for collecting & labeling data from live camera feed☆43Updated last year
- MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in PyTorch. Out-of-box support for retraining on Open Images dataset. ONN…☆55Updated last year
- MonoDepth Estimation - ICRA 2018 "Sparse-to-Dense: Depth Prediction from Sparse Depth Samples and a Single Image"☆13Updated 5 years ago
- Transfer COCO data set annotations to Darknet YOLO annotations format. Hence, Dark(net) Chocolate(COCO)!☆34Updated 5 years ago
- Easy to use neural networks for NVIDIA Jetson (and desktop too!)☆73Updated 2 years ago
- ☆127Updated 2 years ago
- Image classification using pytorch☆59Updated last year
- 3D ResNet Video Classification accelerated by TensorRT☆46Updated 3 years ago
- Hermes is a Wildfire detection system that utilizes Computer Vision and is accelerated using NVIDIA Deepstream☆65Updated 4 years ago
- YOLOv4 accelerated wtih TensorRT and multi-stream input using Deepstream☆36Updated 4 years ago
- Learn how to train a custom instance segmentation model with Detectron2☆100Updated 3 years ago
- Nvidia Deepstream YOLO Python App☆22Updated 4 years ago
- ☆19Updated 2 years ago
- Fast Object Detector for the Jetson Nano☆101Updated 5 years ago
- Dockerfile for yolov5 inference on NVIDIA Jetson☆44Updated 2 years ago
- Train a Mask R-CNN model with the Tensorflow Object Detection API☆77Updated 3 years ago
- Multi class multi instance tracking . You can also perform custom multi object tracking☆15Updated 2 years ago
- Aerial Drone Image Semantic Segmentation using U-Net with MobileNetV2 as the backbone architecture in Pytorch☆12Updated 4 years ago
- How to run Keras model on Jetson Nano☆67Updated 5 years ago
- Unofficial Implementation of ECANets (CVPR 2020) for the Reproducibility Challenge 2020.☆31Updated 3 years ago
- Fever screening with IR & RGB cameras and Deep CNNs☆159Updated 2 years ago
- Predict the status of a genetic biomarker important for brain cancer treatment☆38Updated 2 months ago
- Matthews Correlation Coefficient Loss implementation for image segmentation.☆12Updated last month
- This example demonstrates the use of U-net model for retinal blood vessel segmentation on tensorflow 2.x☆17Updated 4 years ago
- Social Distancing Detector using deep learning and capable to run on edge AI devices such as NVIDIA Jetson, Google Coral, and more.☆141Updated 2 years ago
- Predicting COVID-19 Pneumonia Severity on Chest X-ray with Deep Learning☆26Updated last year
- Fully supervised binary classification of skin lesions from dermatoscopic images using an ensemble of diverse CNN architectures (Efficien…☆47Updated 4 years ago
- Creating custom object detection datasets in the formats of COCO and KITTI☆53Updated 5 years ago