add max dets

dataset/build.py

@@ -8,6 +8,7 @@ try:
     from .data_augment.yolov5_augment import YOLOv5Augmentation, YOLOv5BaseTransform
 
 except:
+    print(999)
     from voc import VOCDetection
     from coco import COCODataset
     from ourdataset import OurDataset