created by : Abegail Lagar
Tools used:
Pytorch Lightning
Wandb
.
├── weights # Pretrained weights
├── outputs # Annotated output
├── samples # Raw video
├── datasets # Dataset folder
│ └── python
│ └── drinks # Drinks Dataset
├── various codes #train ,test codes and other utilities
└── README.md
steps before training and testing the model
!git clone https://github.com/abbylagar/objectdetection.git
%cd <project_folder>
!pip install -r requirements.txt
Datasets and pretrained model during this part.
Sample run in the Drinks_detection.ipynb
!python3 test.py
Output metrics:
Sample run in the Drinks_detection.ipynb .
If wandb visualization prompts , enter 3 (No visualization)
!python3 train.py
Output metrics after 50 epochs
Sample videos with object detection tracking is in the outputs folder Filename : output1.mp4
!python3 infer_record.py