Sendeky / PyTorch-ResNet50-Object-Detection

A PyTorch Object Detector that is based on ResNet50. Loosely following this article: "https://pyimagesearch.com/2021/11/01/training-an-object-detector-from-scratch-in-pytorch/"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

PyTorch-ResNet50-Object-Detection

TLDR:

  • I am trying to understand how to make a Object Detector in PyTorch.
  • I decided to use the KITTI and BDD100k datasets to train it on object detection.
  • In the future I plan to make a HydraNet (paper and article) based off of this that does object detection and depth estimation.

About

A PyTorch Object Detector that is based on ResNet50. Loosely following this article: "https://pyimagesearch.com/2021/11/01/training-an-object-detector-from-scratch-in-pytorch/"


Languages

Language:Python 100.0%