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% |