Yolo pytorch implementation using Resnet18 backbone. NN for multi bboxes detection and classification.
- numpy
- matplotlib
- cv2
- math
- tqdm
- os
- xml.etree.ElementTree
- pandas
- albumentations
- time
- datetime, os
- pickle
There are three different notebooks in this repository:
- inference_on_pets.ipynb : Inference on Resnet18 backbone to test its current capability
- yolo.ipynb : Implementation of YoloV1 network using Resnet18 as backbone
- inference_on_video.ipynb : using YoloV1 to inference on video.