https://github.com/leoneversberg/object-detection-pytorch

A short jupyter notebook to train a Faster R-CNN object detection model with PyTorch v2 using the COCO data format
https://github.com/leoneversberg/object-detection-pytorch

coco deep-learning faster-rcnn microsoft-coco object-detection python pytorch

Last synced: 5 months ago
JSON representation

A short jupyter notebook to train a Faster R-CNN object detection model with PyTorch v2 using the COCO data format

• Host: GitHub
• URL: https://github.com/leoneversberg/object-detection-pytorch
• Owner: leoneversberg
• License: mit
• Created: 2023-12-21T10:58:50.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2024-02-21T10:16:07.000Z (over 2 years ago)
• Last Synced: 2024-02-21T11:29:38.170Z (over 2 years ago)
• Topics: coco, deep-learning, faster-rcnn, microsoft-coco, object-detection, python, pytorch
• Language: Jupyter Notebook
• Homepage: https://medium.com/towards-artificial-intelligence/how-to-train-a-custom-faster-rcnn-model-in-pytorch-128bffa7a014
• Size: 12.6 MB
• Stars: 4
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# object-detection-pytorch

A short jupyter notebook to train a Faster R-CNN object detection model with PyTorch v2.1. 

I used [blender-gen](https://github.com/ignc-research/blender-gen) to create a synthetic image dataset in the Microsoft COCO format.

![loss](./media/loss.png)

![result](./media/result.png)