Training with CSV dataset

VDIGPKU / M2Det

M2Det: A Single-Shot Object Detector based on Multi-Level Feature Pyramid Network

MIT License

1.45k stars 318 forks source link

Training with CSV dataset #108

Closed sudohainguyen closed 2 years ago

sudohainguyen commented 4 years ago

In this PR, I created CustomDataset class which makes our source code now is able to train on custom data with csv files. I also include CSV examples. Moreover, I fixed some bugs from resnet backbone.

devendraswamy commented 4 years ago

HI , what is the custom dataset format ?

and i had dataset like .csv format and is like in the form of """" FILENAME , XMIN , YMIN , XMAX , YMIN , CLASS . is it work for this format ??

kindly help me !!

Thanking you !!

devendraswamy commented 4 years ago

AttributeError: 'CustomDataset' object has no attribute 'evaluate_detections' please help me to solve this error