diff --git a/configs/picodet/legacy_model/application/layout_analysis/picodet_lcnet_x1_0_layout.yml b/configs/picodet/legacy_model/application/layout_analysis/picodet_lcnet_x1_0_layout.yml index b4bec58d7a9d036c943b67e6906bfa6d439ab052..db9a8c683adc74ed043472d3d0e8fbb4c3b537f5 100644 --- a/configs/picodet/legacy_model/application/layout_analysis/picodet_lcnet_x1_0_layout.yml +++ b/configs/picodet/legacy_model/application/layout_analysis/picodet_lcnet_x1_0_layout.yml @@ -26,14 +26,14 @@ metric: COCO num_classes: 5 TrainDataset: - !COCODataSet + name: COCODataSet image_dir: train anno_path: train.json dataset_dir: ./dataset/publaynet/ data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd'] EvalDataset: - !COCODataSet + name: COCODataSet image_dir: val anno_path: val.json dataset_dir: ./dataset/publaynet/