diff --git a/configs/picodet/legacy_model/application/mainbody_detection/picodet_lcnet_x2_5_640_mainbody.yml b/configs/picodet/legacy_model/application/mainbody_detection/picodet_lcnet_x2_5_640_mainbody.yml index a948565f9fe6b589c0c39ecbc0a3768846f77b31..5b4d9769f093a008a9367d79e608f49d25c722c3 100644 --- a/configs/picodet/legacy_model/application/mainbody_detection/picodet_lcnet_x2_5_640_mainbody.yml +++ b/configs/picodet/legacy_model/application/mainbody_detection/picodet_lcnet_x2_5_640_mainbody.yml @@ -1,4 +1,5 @@ _BASE_: [ + '../../../../datasets/coco_detection.yml', '../../../../runtime.yml', '../../_base_/picodet_esnet.yml', '../../_base_/optimizer_100e.yml', @@ -20,23 +21,3 @@ PicoDet: LCNet: scale: 2.5 feature_maps: [3, 4, 5] - -metric: COCO -num_classes: 1 - -TrainDataset: - !COCODataSet - image_dir: ./ - anno_path: train.json - dataset_dir: dataset/mainbody/ - data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd'] - -EvalDataset: - !COCODataSet - image_dir: ./ - anno_path: val.json - dataset_dir: dataset/mainbody/ - -TestDataset: - !ImageFolder - anno_path: ./dataset/mainbody/val.json