版面分析

configs/datasets/coco_V1_3_27K.yml

@@ -0,0 +1,20 @@
+metric: COCO
+num_classes: 3
+
+TrainDataset:
+  !COCODataSet
+    image_dir: train/images
+    anno_path: annotations/instances_train.json
+    dataset_dir: Layout_Analysis/V1.3_27K
+    data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
+
+EvalDataset:
+  !COCODataSet
+    image_dir: eval/images
+    anno_path: annotations/instances_val.json
+    dataset_dir: Layout_Analysis/V1.3_27K
+
+TestDataset:
+  !ImageFolder
+    anno_path: annotations/instances_val.json # also support txt (like VOC's label_list.txt)
+    dataset_dir: Layout_Analysis/V1.3_27K # if set, anno_path will be 'dataset_dir/anno_path'

configs/picodet/picodet_l_320_coco_lcnet_V1_3_27K_20230201.yml

@@ -0,0 +1,45 @@
+_BASE_: [
+  '../datasets/coco_V1_3_27K.yml',
+  '../runtime.yml',
+  '_base_/picodet_v2.yml',
+  '_base_/optimizer_300e.yml',
+  '_base_/picodet_320_reader.yml',
+]
+
+pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/PPLCNet_x2_0_pretrained.pdparams
+weights: output/picodet_l_320_coco/best_model
+find_unused_parameters: True
+use_ema: true
+epoch: 250
+snapshot_epoch: 10
+
+LCNet:
+  scale: 2.0
+  feature_maps: [3, 4, 5]
+
+LCPAN:
+  out_channels: 160
+
+PicoHeadV2:
+  conv_feat:
+    name: PicoFeat
+    feat_in: 160
+    feat_out: 160
+    num_convs: 4
+    num_fpn_stride: 4
+    norm_type: bn
+    share_cls_reg: True
+    use_se: True
+  feat_in_chan: 160
+
+LearningRate:
+  base_lr: 0.12
+  schedulers:
+  - !CosineDecay
+    max_epochs: 300
+  - !LinearWarmup
+    start_factor: 0.1
+    steps: 300
+
+TrainReader:
+  batch_size: 24

configs/picodet/picodet_l_416_coco_lcnet_V1_3_27K_20230201.yml

@@ -0,0 +1,45 @@
+_BASE_: [
+  '../datasets/coco_V1_3_27K.yml',
+  '../runtime.yml',
+  '_base_/picodet_v2.yml',
+  '_base_/optimizer_300e.yml',
+  '_base_/picodet_416_reader.yml',
+]
+
+pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/PPLCNet_x2_0_pretrained.pdparams
+weights: output/picodet_l_320_coco/best_model
+find_unused_parameters: True
+use_ema: true
+epoch: 250
+snapshot_epoch: 10
+
+LCNet:
+  scale: 2.0
+  feature_maps: [3, 4, 5]
+
+LCPAN:
+  out_channels: 160
+
+PicoHeadV2:
+  conv_feat:
+    name: PicoFeat
+    feat_in: 160
+    feat_out: 160
+    num_convs: 4
+    num_fpn_stride: 4
+    norm_type: bn
+    share_cls_reg: True
+    use_se: True
+  feat_in_chan: 160
+
+LearningRate:
+  base_lr: 0.12
+  schedulers:
+  - !CosineDecay
+    max_epochs: 300
+  - !LinearWarmup
+    start_factor: 0.1
+    steps: 300
+
+TrainReader:
+  batch_size: 24

configs/picodet/picodet_l_640_coco_lcnet_V1_3_27K_20230201.yml

@@ -0,0 +1,45 @@
+_BASE_: [
+  '../datasets/coco_V1_3_27K.yml',
+  '../runtime.yml',
+  '_base_/picodet_v2.yml',
+  '_base_/optimizer_300e.yml',
+  '_base_/picodet_640_reader.yml',
+]
+
+pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/PPLCNet_x2_0_pretrained.pdparams
+weights: output/picodet_l_320_coco/best_model
+find_unused_parameters: True
+use_ema: true
+epoch: 200
+snapshot_epoch: 10
+
+LCNet:
+  scale: 2.0
+  feature_maps: [3, 4, 5]
+
+LCPAN:
+  out_channels: 160
+
+PicoHeadV2:
+  conv_feat:
+    name: PicoFeat
+    feat_in: 160
+    feat_out: 160
+    num_convs: 4
+    num_fpn_stride: 4
+    norm_type: bn
+    share_cls_reg: True
+    use_se: True
+  feat_in_chan: 160
+
+LearningRate:
+  base_lr: 0.06
+  schedulers:
+  - !CosineDecay
+    max_epochs: 300
+  - !LinearWarmup
+    start_factor: 0.1
+    steps: 300
+
+TrainReader:
+  batch_size: 12

configs/picodet/picodet_m_320_coco_lcnet_V1_3_27K_20230201.yml

@@ -0,0 +1,25 @@
+_BASE_: [
+  '../datasets/coco_V1_3_27K.yml',
+  '../runtime.yml',
+  '_base_/picodet_v2.yml',
+  '_base_/optimizer_300e.yml',
+  '_base_/picodet_320_reader.yml',
+]
+
+weights: output/picodet_m_320_coco/best_model
+find_unused_parameters: True
+use_ema: true
+epoch: 300
+snapshot_epoch: 10
+
+TrainReader:
+  batch_size: 48
+
+LearningRate:
+  base_lr: 0.24
+  schedulers:
+  - !CosineDecay
+    max_epochs: 300
+  - !LinearWarmup
+    start_factor: 0.1
+    steps: 300

configs/picodet/picodet_m_416_coco_lcnet_V1_3_27K_20230201.yml

@@ -0,0 +1,25 @@
+_BASE_: [
+  '../datasets/coco_V1_3_27K.yml',
+  '../runtime.yml',
+  '_base_/picodet_v2.yml',
+  '_base_/optimizer_300e.yml',
+  '_base_/picodet_416_reader.yml',
+]
+
+weights: output/picodet_m_416_coco/best_model
+find_unused_parameters: True
+use_ema: true
+epoch: 250
+snapshot_epoch: 10
+
+TrainReader:
+  batch_size: 48
+
+LearningRate:
+  base_lr: 0.24
+  schedulers:
+  - !CosineDecay
+    max_epochs: 300
+  - !LinearWarmup
+    start_factor: 0.1
+    steps: 300