dusty-nv / pytorch-classificationLinks
Training of image classification models with PyTorch
☆33Updated 2 years ago
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☆34Updated 5 years ago
- Learn how to train a custom instance segmentation model with Detectron2☆102Updated 4 years ago
- Object Detection and Image Segmentation with Detectron2☆164Updated 3 years ago
- MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in PyTorch. Out-of-box support for retraining on Open Images dataset. ONN…☆55Updated 2 years ago
- How to create custom COCO data set for instance segmentation☆195Updated 4 years ago
- Transfer COCO data set annotations to Darknet YOLO annotations format. Hence, Dark(net) Chocolate(COCO)!☆34Updated 6 years ago
- Train a Mask R-CNN model with the Tensorflow Object Detection API☆76Updated 4 years ago
- Packaged version of ultralytics/yolov5 + many extra features☆294Updated 9 months ago
- An easy way to train a Yolo object detector with your custom images dataset, using PyTorch☆83Updated 6 years ago
- Train your own custom MaskRCNN Object Detection and Instance Segmentation model.☆62Updated 4 years ago
- Learning and Building Convolutional Neural Networks using PyTorch☆218Updated 3 years ago
- ☆126Updated 3 years ago
- YOLO v5 pytorch implementation. This repository contains code for this tutorial.☆151Updated 2 years ago
- Calculate mean Average Precision (mAP) and confusion matrix for object detection models. Bounding box information for groundtruth and pre…☆39Updated 6 years ago
- ☆86Updated last year
- vizgradcam is the fastest way to visualize GradCAM with your Keras models.☆60Updated 4 years ago
- Minimal PyTorch implementation of YOLOv4☆33Updated 2 years ago
- Tensorflow implementation of DETR : Object Detection with Transformers☆171Updated 3 years ago
- GUI tool for collecting & labeling data from live camera feed☆43Updated last year
- This script converts FLIR thermal dataset annotations to YOLO format☆41Updated 5 years ago
- ☆84Updated 3 years ago
- Python scripts performing object detection using the YOLOv6 model in ONNX.☆96Updated 3 years ago
- label file converter: xml to yolo☆74Updated 5 years ago
- A semantic segmentation pipeline for custom image annotation☆133Updated 3 years ago
- A multibranch neural net architecture. (PyTorch)☆90Updated 5 years ago
- Convert VOC format XMLs to COCO format json☆102Updated 5 years ago
- Tensorflow 2 Object Detection API Tutorial. This tutorial will take you from installation, to running pre-trained detection model, and tr…☆115Updated 5 years ago
- Exporting yolov5 to tflite and run inference on raspberry pi and cpu☆56Updated 3 years ago
- A clean, modular implementation of the Yolov7 model family, which uses the official pretrained weights, with utilities for training the m…☆114Updated last year
- Creating custom object detection datasets in the formats of COCO and KITTI☆53Updated 6 years ago