-Residual Neural Network Object Detector written for Pycocotool's library. Model implements custom skip block connections and uses a custom dataset loader for image classification object detection. This repository consists of 3 files, MSE, IOU and Val. The first two, MSE and IoU are models that use different loss metrics for backpropagation; with MSE using Mean Square Error for back propagation and IoU using intersection over union. Running these two models will produce two separate neural nets, which can be further trained. The Val file takes these two models (saved as mse_net and iou_net) and evaluates its performance through data visualization of confusion matrices and producing randomly sampled images and indicates their predicted class labels and true labels. Note that the code requires two files, an annotations file containing the .json annotations for the data and a training dataset whose labels correspond to the annotations file. This can be downloaded from the Pycocotools website; the data used in these models were the train and val 2014 datasets.
0 commit comments