dusty-nv / pytorch-classification
Training of image classification models with PyTorch
☆31Updated last year
Alternatives and similar repositories for pytorch-classification:
Users that are interested in pytorch-classification are comparing it to the libraries listed below
- Training of semantic segmentation networks with PyTorch☆34Updated 4 years ago
- GUI tool for collecting & labeling data from live camera feed☆42Updated 9 months ago
- MonoDepth Estimation - ICRA 2018 "Sparse-to-Dense: Depth Prediction from Sparse Depth Samples and a Single Image"☆12Updated 5 years ago
- MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in PyTorch. Out-of-box support for retraining on Open Images dataset. ONN…☆54Updated last year
- Object Detection and Image Segmentation with Detectron2☆158Updated 2 years ago
- Transfer COCO data set annotations to Darknet YOLO annotations format. Hence, Dark(net) Chocolate(COCO)!☆34Updated 5 years ago
- Learn how to train a custom instance segmentation model with Detectron2☆99Updated 3 years ago
- Train your own custom MaskRCNN Object Detection and Instance Segmentation model.☆63Updated 3 years ago
- Implementing polyp segmentation using the U-Net and CVC-612 dataset.☆67Updated 3 years ago
- A (mostly from scratch) PyTorch implementation of YOLOv3 for real-time webcam object detection☆21Updated 4 years ago
- ☆19Updated 2 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 3 years ago
- PyTorch Lightning based training of Semantic Segmentation models☆38Updated 2 years ago
- vizgradcam is the fastest way to visualize GradCAM with your Keras models.☆60Updated 3 years ago
- Easy to use neural networks for NVIDIA Jetson (and desktop too!)☆74Updated 2 years ago
- Minimal PyTorch implementation of YOLOv4☆33Updated last year
- Experiments with UNET/FPN models and cityscapes/kitti datasets [Pytorch]☆122Updated 8 months ago
- Transfer Learning with EfficientNet in Keras☆88Updated last year
- How to run Keras model on Jetson Nano☆67Updated 5 years ago
- Creating custom object detection datasets in the formats of COCO and KITTI☆52Updated 5 years ago
- Minimal PyTorch implementation of YOLOv5 and StrongSort☆67Updated 2 years ago
- Tensorflow 2.x implementation YOLACT☆33Updated 2 years ago
- Fast Object Detector for the Jetson Nano☆102Updated 5 years ago
- A Keras implementation of hybrid efficientnet swin transformer model.☆33Updated last year
- ☆16Updated 8 months ago
- Tensorflow 2.0 Implementation of GCViT: Global Context Vision Transformer☆27Updated last year
- Building, training and deploying a fully convolutional neural network in TensorFlow☆40Updated 5 years ago
- How to run SSD Mobilenet V2 object detection on Jetson Nano at 20+ FPS☆33Updated 5 years ago
- ☆126Updated 2 years ago