This repository contains the YOLO model trained on the COCO Dataset for Object Detection. The model gave an IoU of about 69% on the test dataset and can be used as a onnx model
Model Details:
- Task: Object Detection
- Architecture : YOLO Model stored in onnx format
- Input Size: Varying size but resized to 640x640 pixels with 3 channels(RGB)
- Dataset : The model is trained on unnormalized dataset
- Dataset used: PASCAL VOC Dataset which is a subset of COCO Dataset
- IoU: 69%