Object-Detection

Python Version - Python 3.10

pip install -r requirements.txt

Train images must be placed in: images directory

Train labels must be placed in: labels directory

python train.py

python inference.py

Trained model is located in: https://drive.google.com/drive/folders/1DAeYLMtmMXdzRHGvuuw8TLeVXJnkaYBb?usp=sharing

Download it into object_detection_component directory

Since the size of this model is big, we are not adding it to zip file.

After downloading, the model will present in the filepath: object_detection_component/fasterrcnn_resnet50_fpn_custom.pth

colab_notebook_train_inference_plotting.ipynb