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%