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
- Learn how to train a custom instance segmentation model with Detectron2☆100Updated 3 years ago
- GUI tool for collecting & labeling data from live camera feed☆43Updated 11 months ago
- MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in PyTorch. Out-of-box support for retraining on Open Images dataset. ONN…☆55Updated last year
- Easy to use neural networks for NVIDIA Jetson (and desktop too!)☆73Updated 2 years ago
- YOLOv4 accelerated wtih TensorRT and multi-stream input using Deepstream☆36Updated 4 years ago
- Calculate mean Average Precision (mAP) and confusion matrix for object detection models. Bounding box information for groundtruth and pre…☆39Updated 5 years ago
- Dockerfile for yolov5 inference on NVIDIA Jetson☆44Updated 2 years ago
- Training of object detection networks with PyTorch☆16Updated last year
- label file converter: xml to yolo☆72Updated 5 years ago
- Minimal PyTorch implementation of YOLOv5 and StrongSort☆67Updated 2 years ago
- Object detection and instance segmentation on MaskRCNN with torchvision, albumentations, tensorboard and cocoapi. Supports custom coco da…☆19Updated 4 years ago
- A (mostly from scratch) PyTorch implementation of YOLOv3 for real-time webcam object detection☆21Updated 4 years ago
- Implementing polyp segmentation using the U-Net and CVC-612 dataset.☆67Updated 3 years ago
- Nvidia Deepstream YOLO Python App☆22Updated 4 years ago
- Train a Mask R-CNN model with the Tensorflow Object Detection API☆77Updated 3 years ago
- Run YOLOv4 directly with OpenCV using the CUDA enabled DNN module.☆37Updated 4 years ago
- This Repo covers all formats of annotations for Object Detection and can easily convert from one form to another using attached scripts☆34Updated 4 years ago
- MonoDepth Estimation - ICRA 2018 "Sparse-to-Dense: Depth Prediction from Sparse Depth Samples and a Single Image"☆13Updated 5 years ago
- ☆19Updated 2 years ago
- A personal implementation of YOLOv5 (v6.0)☆52Updated last year
- Vision Transformer using TensorFlow 2.0☆98Updated 4 years ago
- Aerial Drone Image Semantic Segmentation using U-Net with MobileNetV2 as the backbone architecture in Pytorch☆12Updated 4 years ago
- This example demonstrates the use of U-net model for retinal blood vessel segmentation on tensorflow 2.x☆17Updated 4 years ago
- tools for converting yolo boxes to pascal voc xml and TFRecords☆33Updated 5 years ago
- Convert LabelMe Annotation Tool JSON format to YOLO text file format☆58Updated 2 years ago
- Pytorch Implementation of Single Shot MultiBox Detector (SSD)☆23Updated 3 years ago
- ☆15Updated 3 years ago
- vizgradcam is the fastest way to visualize GradCAM with your Keras models.☆60Updated 3 years ago
- 3D ResNet Video Classification accelerated by TensorRT☆46Updated 3 years ago