├── .gitignore ├── README.md ├── README_cn.md ├── README_en.md ├── configs ├── cascade_rcnn │ ├── README.md │ ├── _base_ │ │ ├── cascade_fpn_reader.yml │ │ ├── cascade_mask_fpn_reader.yml │ │ ├── cascade_mask_rcnn_r50_fpn.yml │ │ ├── cascade_rcnn_r50_fpn.yml │ │ └── optimizer_1x.yml │ ├── cascade_mask_rcnn_r50_fpn_1x_coco.yml │ ├── cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco.yml │ ├── cascade_mask_rcnn_r50_vd_fpn_ssld_2x_coco.yml │ ├── cascade_rcnn_r50_fpn_1x_coco.yml │ ├── cascade_rcnn_r50_vd_fpn_ssld_1x_coco.yml │ └── cascade_rcnn_r50_vd_fpn_ssld_2x_coco.yml ├── centernet │ ├── README.md │ ├── README_cn.md │ ├── _base_ │ │ ├── centernet_dla34.yml │ │ ├── centernet_r50.yml │ │ ├── centernet_reader.yml │ │ └── optimizer_140e.yml │ ├── centernet_dla34_140e_coco.yml │ ├── centernet_mbv1_140e_coco.yml │ ├── centernet_mbv3_large_140e_coco.yml │ ├── centernet_mbv3_small_140e_coco.yml │ ├── centernet_r50_140e_coco.yml │ └── centernet_shufflenetv2_140e_coco.yml ├── convnext │ ├── README.md │ ├── ppyoloe_convnext_tiny_36e_coco.yml │ ├── yolov5_convnext_s_36e_coco.yml │ └── yolox_convnext_s_36e_coco.yml ├── datasets │ ├── coco_detection.yml │ ├── coco_instance.yml │ ├── dota.yml │ ├── dota_ms.yml │ ├── mcmot.yml │ ├── mot.yml │ ├── objects365_detection.yml │ ├── roadsign_voc.yml │ ├── sniper_coco_detection.yml │ ├── sniper_visdrone_detection.yml │ ├── spine_coco.yml │ ├── visdrone_detection.yml │ ├── voc.yml │ └── wider_face.yml ├── dcn │ ├── README.md │ ├── cascade_rcnn_dcn_r50_fpn_1x_coco.yml │ ├── cascade_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml │ ├── faster_rcnn_dcn_r101_vd_fpn_1x_coco.yml │ ├── faster_rcnn_dcn_r50_fpn_1x_coco.yml │ ├── faster_rcnn_dcn_r50_vd_fpn_1x_coco.yml │ ├── faster_rcnn_dcn_r50_vd_fpn_2x_coco.yml │ ├── faster_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml │ ├── mask_rcnn_dcn_r101_vd_fpn_1x_coco.yml │ ├── mask_rcnn_dcn_r50_fpn_1x_coco.yml │ ├── mask_rcnn_dcn_r50_vd_fpn_2x_coco.yml │ └── mask_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml ├── deformable_detr │ ├── README.md │ ├── _base_ │ │ ├── deformable_detr_r50.yml │ │ ├── deformable_detr_reader.yml │ │ └── deformable_optimizer_1x.yml │ └── deformable_detr_r50_1x_coco.yml ├── detr │ ├── README.md │ ├── _base_ │ │ ├── detr_r50.yml │ │ ├── detr_reader.yml │ │ └── optimizer_1x.yml │ └── detr_r50_1x_coco.yml ├── dino │ ├── README.md │ ├── _base_ │ │ ├── dino_focalnet.yml │ │ ├── dino_r50.yml │ │ ├── dino_reader.yml │ │ ├── dino_swin.yml │ │ ├── optimizer_1x.yml │ │ ├── optimizer_2x.yml │ │ └── optimizer_3x.yml │ ├── dino_r50_4scale_1x_coco.yml │ └── dino_r50_4scale_2x_coco.yml ├── face_detection │ ├── README.md │ ├── README_en.md │ ├── _base_ │ │ ├── blazeface.yml │ │ ├── blazeface_fpn.yml │ │ ├── face_reader.yml │ │ └── optimizer_1000e.yml │ ├── blazeface_1000e.yml │ └── blazeface_fpn_ssh_1000e.yml ├── faster_rcnn │ ├── README.md │ ├── _base_ │ │ ├── faster_fpn_reader.yml │ │ ├── faster_rcnn_r50.yml │ │ ├── faster_rcnn_r50_fpn.yml │ │ ├── faster_rcnn_swin_reader.yml │ │ ├── faster_rcnn_swin_tiny_fpn.yml │ │ ├── faster_reader.yml │ │ ├── optimizer_1x.yml │ │ └── optimizer_swin_1x.yml │ ├── faster_rcnn_r101_1x_coco.yml │ ├── faster_rcnn_r101_fpn_1x_coco.yml │ ├── faster_rcnn_r101_fpn_2x_coco.yml │ ├── faster_rcnn_r101_vd_fpn_1x_coco.yml │ ├── faster_rcnn_r101_vd_fpn_2x_coco.yml │ ├── faster_rcnn_r34_fpn_1x_coco.yml │ ├── faster_rcnn_r34_fpn_multiscaletest_1x_coco.yml │ ├── faster_rcnn_r34_vd_fpn_1x_coco.yml │ ├── faster_rcnn_r50_1x_coco.yml │ ├── faster_rcnn_r50_fpn_1x_coco.yml │ ├── faster_rcnn_r50_fpn_2x_coco.yml │ ├── faster_rcnn_r50_vd_1x_coco.yml │ ├── faster_rcnn_r50_vd_fpn_1x_coco.yml │ ├── faster_rcnn_r50_vd_fpn_2x_coco.yml │ ├── faster_rcnn_r50_vd_fpn_ssld_1x_coco.yml │ ├── faster_rcnn_r50_vd_fpn_ssld_2x_coco.yml │ ├── faster_rcnn_swin_tiny_fpn_1x_coco.yml │ ├── faster_rcnn_swin_tiny_fpn_2x_coco.yml │ ├── faster_rcnn_swin_tiny_fpn_3x_coco.yml │ ├── faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml │ └── faster_rcnn_x101_vd_64x4d_fpn_2x_coco.yml ├── fcos │ ├── README.md │ ├── _base_ │ │ ├── fcos_r50_fpn.yml │ │ ├── fcos_reader.yml │ │ └── optimizer_1x.yml │ ├── fcos_dcn_r50_fpn_1x_coco.yml │ ├── fcos_r50_fpn_1x_coco.yml │ ├── fcos_r50_fpn_iou_1x_coco.yml │ ├── fcos_r50_fpn_iou_multiscale_2x_coco.yml │ └── fcos_r50_fpn_multiscale_2x_coco.yml ├── few-shot │ ├── README.md │ ├── _base_ │ │ ├── faster_fpn_reader.yml │ │ ├── faster_rcnn_r50.yml │ │ ├── faster_rcnn_r50_fpn.yml │ │ ├── faster_reader.yml │ │ ├── optimizer_1x.yml │ │ ├── optimizer_80e.yml │ │ ├── ppyoloe_plus_crn.yml │ │ └── ppyoloe_plus_reader.yml │ ├── faster_rcnn_r50_vd_fpn_1x_coco_cotuning_roadsign.yml │ └── ppyoloe_plus_crn_s_80e_contrast_pcb.yml ├── focalnet │ ├── README.md │ └── ppyoloe_plus_focalnet_tiny_36e_coco.yml ├── gfl │ ├── README.md │ ├── _base_ │ │ ├── gfl_r50_fpn.yml │ │ ├── gfl_reader.yml │ │ ├── gflv2_r50_fpn.yml │ │ └── optimizer_1x.yml │ ├── gfl_r101vd_fpn_mstrain_2x_coco.yml │ ├── gfl_r18vd_1x_coco.yml │ ├── gfl_r34vd_1x_coco.yml │ ├── gfl_r50_fpn_1x_coco.yml │ ├── gfl_slim_ld_r18vd_1x_coco.yml │ └── gflv2_r50_fpn_1x_coco.yml ├── gn │ ├── README.md │ ├── cascade_mask_rcnn_r50_fpn_gn_2x_coco.yml │ ├── cascade_rcnn_r50_fpn_gn_2x_coco.yml │ ├── faster_rcnn_r50_fpn_gn_2x_coco.yml │ └── mask_rcnn_r50_fpn_gn_2x_coco.yml ├── group_detr │ ├── README.md │ ├── _base_ │ │ ├── dino_2000_reader.yml │ │ ├── dino_reader.yml │ │ ├── group_dino_r50.yml │ │ ├── group_dino_vit_huge.yml │ │ └── optimizer_1x.yml │ ├── group_dino_r50_4scale_1x_coco.yml │ └── group_dino_vit_huge_4scale_1x_coco.yml ├── hrnet │ ├── README.md │ ├── _base_ │ │ └── faster_rcnn_hrnetv2p_w18.yml │ ├── faster_rcnn_hrnetv2p_w18_1x_coco.yml │ └── faster_rcnn_hrnetv2p_w18_2x_coco.yml ├── keypoint │ ├── KeypointBenchmark.md │ ├── README.md │ ├── README_en.md │ ├── higherhrnet │ │ ├── higherhrnet_hrnet_w32_512.yml │ │ ├── higherhrnet_hrnet_w32_512_swahr.yml │ │ └── higherhrnet_hrnet_w32_640.yml │ ├── hrnet │ │ ├── dark_hrnet_w32_256x192.yml │ │ ├── dark_hrnet_w32_384x288.yml │ │ ├── dark_hrnet_w48_256x192.yml │ │ ├── hrnet_w32_256x192.yml │ │ ├── hrnet_w32_256x256_mpii.yml │ │ └── hrnet_w32_384x288.yml │ ├── lite_hrnet │ │ ├── lite_hrnet_18_256x192_coco.yml │ │ ├── lite_hrnet_18_384x288_coco.yml │ │ ├── lite_hrnet_30_256x192_coco.yml │ │ ├── lite_hrnet_30_384x288_coco.yml │ │ └── wider_naive_hrnet_18_256x192_coco.yml │ ├── petr │ │ └── petr_resnet50_16x2_coco.yml │ ├── tiny_pose │ │ ├── README.md │ │ ├── README_en.md │ │ ├── tinypose_128x96.yml │ │ └── tinypose_256x192.yml │ └── vit_pose │ │ ├── vitpose_base_coco_256x196.yml │ │ └── vitpose_base_simple_coco_256x192.yml ├── mask_rcnn │ ├── README.md │ ├── _base_ │ │ ├── mask_fpn_reader.yml │ │ ├── mask_rcnn_r50.yml │ │ ├── mask_rcnn_r50_fpn.yml │ │ ├── mask_reader.yml │ │ └── optimizer_1x.yml │ ├── mask_rcnn_r101_fpn_1x_coco.yml │ ├── mask_rcnn_r101_vd_fpn_1x_coco.yml │ ├── mask_rcnn_r50_1x_coco.yml │ ├── mask_rcnn_r50_2x_coco.yml │ ├── mask_rcnn_r50_fpn_1x_coco.yml │ ├── mask_rcnn_r50_fpn_2x_coco.yml │ ├── mask_rcnn_r50_vd_fpn_1x_coco.yml │ ├── mask_rcnn_r50_vd_fpn_2x_coco.yml │ ├── mask_rcnn_r50_vd_fpn_ssld_1x_coco.yml │ ├── mask_rcnn_r50_vd_fpn_ssld_2x_coco.yml │ ├── mask_rcnn_x101_vd_64x4d_fpn_1x_coco.yml │ └── mask_rcnn_x101_vd_64x4d_fpn_2x_coco.yml ├── mot │ ├── DataDownload.md │ ├── README.md │ ├── README_en.md │ ├── botsort │ │ ├── README.md │ │ ├── README_cn.md │ │ └── botsort_ppyoloe.yml │ ├── bytetrack │ │ ├── README.md │ │ ├── README_cn.md │ │ ├── _base_ │ │ │ ├── ht21.yml │ │ │ ├── mix_det.yml │ │ │ ├── mix_mot_ch.yml │ │ │ ├── mot17.yml │ │ │ ├── ppyoloe_mot_reader_640x640.yml │ │ │ ├── yolov3_mot_reader_608x608.yml │ │ │ └── yolox_mot_reader_800x1440.yml │ │ ├── bytetrack_ppyoloe.yml │ │ ├── bytetrack_ppyoloe_pplcnet.yml │ │ ├── bytetrack_ppyoloe_ppvehicle9cls.yml │ │ ├── bytetrack_yolov3.yml │ │ ├── bytetrack_yolox.yml │ │ ├── bytetrack_yolox_ht21.yml │ │ └── detector │ │ │ ├── README.md │ │ │ ├── README_cn.md │ │ │ ├── ppyoloe_crn_l_36e_640x640_mot17half.yml │ │ │ ├── yolov3_darknet53_40e_608x608_mot17half.yml │ │ │ ├── yolox_x_24e_800x1440_ht21.yml │ │ │ ├── yolox_x_24e_800x1440_mix_det.yml │ │ │ └── yolox_x_24e_800x1440_mix_mot_ch.yml │ ├── centertrack │ │ ├── README.md │ │ ├── README_cn.md │ │ ├── _base_ │ │ │ ├── centertrack_dla34.yml │ │ │ ├── centertrack_reader.yml │ │ │ └── optimizer_70e.yml │ │ ├── centertrack_dla34_70e_mot17.yml │ │ └── centertrack_dla34_70e_mot17half.yml │ ├── deepsort │ │ ├── README.md │ │ ├── README_cn.md │ │ ├── _base_ │ │ │ ├── deepsort_reader_1088x608.yml │ │ │ └── mot17.yml │ │ ├── deepsort_jde_yolov3_pcb_pyramid.yml │ │ ├── deepsort_jde_yolov3_pplcnet.yml │ │ ├── deepsort_ppyoloe_pplcnet.yml │ │ ├── deepsort_ppyoloe_resnet.yml │ │ ├── deepsort_ppyolov2_pplcnet.yml │ │ ├── deepsort_yolov3_pplcnet.yml │ │ ├── detector │ │ │ ├── README.md │ │ │ ├── README_cn.md │ │ │ ├── jde_yolov3_darknet53_30e_1088x608_mix.yml │ │ │ ├── ppyoloe_crn_l_36e_640x640_mot17half.yml │ │ │ ├── ppyolov2_r50vd_dcn_365e_640x640_mot17half.yml │ │ │ └── yolov3_darknet53_40e_608x608_mot17half.yml │ │ └── reid │ │ │ ├── README.md │ │ │ ├── README_cn.md │ │ │ ├── deepsort_pcb_pyramid_r101.yml │ │ │ ├── deepsort_pplcnet.yml │ │ │ ├── deepsort_pplcnet_vehicle.yml │ │ │ └── deepsort_resnet.yml │ ├── fairmot │ │ ├── README.md │ │ ├── README_cn.md │ │ ├── _base_ │ │ │ ├── fairmot_dla34.yml │ │ │ ├── fairmot_hardnet85.yml │ │ │ ├── fairmot_hrnetv2_w18_dlafpn.yml │ │ │ ├── fairmot_reader_1088x608.yml │ │ │ ├── fairmot_reader_576x320.yml │ │ │ ├── fairmot_reader_864x480.yml │ │ │ ├── optimizer_30e.yml │ │ │ └── optimizer_30e_momentum.yml │ │ ├── fairmot_dla34_30e_1088x608.yml │ │ ├── fairmot_dla34_30e_1088x608_airplane.yml │ │ ├── fairmot_dla34_30e_1088x608_bytetracker.yml │ │ ├── fairmot_dla34_30e_576x320.yml │ │ ├── fairmot_dla34_30e_864x480.yml │ │ ├── fairmot_enhance_dla34_60e_1088x608.yml │ │ ├── fairmot_enhance_hardnet85_30e_1088x608.yml │ │ ├── fairmot_hrnetv2_w18_dlafpn_30e_1088x608.yml │ │ ├── fairmot_hrnetv2_w18_dlafpn_30e_576x320.yml │ │ └── fairmot_hrnetv2_w18_dlafpn_30e_864x480.yml │ ├── headtracking21 │ │ ├── README.md │ │ ├── README_cn.md │ │ └── fairmot_dla34_30e_1088x608_headtracking21.yml │ ├── jde │ │ ├── README.md │ │ ├── README_cn.md │ │ ├── _base_ │ │ │ ├── jde_darknet53.yml │ │ │ ├── jde_reader_1088x608.yml │ │ │ ├── jde_reader_576x320.yml │ │ │ ├── jde_reader_864x480.yml │ │ │ ├── optimizer_30e.yml │ │ │ └── optimizer_60e.yml │ │ ├── jde_darknet53_30e_1088x608.yml │ │ ├── jde_darknet53_30e_576x320.yml │ │ └── jde_darknet53_30e_864x480.yml │ ├── mcfairmot │ │ ├── README.md │ │ ├── README_cn.md │ │ ├── mcfairmot_dla34_30e_1088x608_visdrone.yml │ │ ├── mcfairmot_dla34_30e_1088x608_visdrone_vehicle_bytetracker.yml │ │ ├── mcfairmot_hrnetv2_w18_dlafpn_30e_1088x608_visdrone.yml │ │ ├── mcfairmot_hrnetv2_w18_dlafpn_30e_1088x608_visdrone_vehicle_bytetracker.yml │ │ ├── mcfairmot_hrnetv2_w18_dlafpn_30e_576x320_bdd100k_mcmot.yml │ │ ├── mcfairmot_hrnetv2_w18_dlafpn_30e_576x320_visdrone.yml │ │ └── mcfairmot_hrnetv2_w18_dlafpn_30e_864x480_visdrone.yml │ ├── mtmct │ │ ├── README.md │ │ ├── README_cn.md │ │ └── gen_aicity_mtmct_data.py │ ├── ocsort │ │ ├── README.md │ │ ├── README_cn.md │ │ ├── ocsort_ppyoloe.yml │ │ └── ocsort_yolox.yml │ ├── pedestrian │ │ ├── README.md │ │ ├── README_cn.md │ │ ├── fairmot_dla34_30e_1088x608_pathtrack.yml │ │ ├── fairmot_dla34_30e_1088x608_visdrone_pedestrian.yml │ │ ├── fairmot_hrnetv2_w18_dlafpn_30e_1088x608_visdrone_pedestrian.yml │ │ ├── fairmot_hrnetv2_w18_dlafpn_30e_576x320_visdrone_pedestrian.yml │ │ ├── fairmot_hrnetv2_w18_dlafpn_30e_864x480_visdrone_pedestrian.yml │ │ └── tools │ │ │ └── visdrone │ │ │ └── visdrone2mot.py │ └── vehicle │ │ ├── README.md │ │ ├── README_cn.md │ │ ├── fairmot_dla34_30e_1088x608_bdd100kmot_vehicle.yml │ │ ├── fairmot_dla34_30e_1088x608_kitti_vehicle.yml │ │ ├── fairmot_dla34_30e_1088x608_visdrone_vehicle.yml │ │ ├── fairmot_hrnetv2_w18_dlafpn_30e_1088x608_visdrone_vehicle.yml │ │ ├── fairmot_hrnetv2_w18_dlafpn_30e_576x320_bdd100kmot_vehicle.yml │ │ ├── fairmot_hrnetv2_w18_dlafpn_30e_576x320_visdrone_vehicle.yml │ │ ├── fairmot_hrnetv2_w18_dlafpn_30e_864x480_visdrone_vehicle.yml │ │ └── tools │ │ ├── bdd100kmot │ │ ├── bdd100k2mot.py │ │ ├── gen_bdd100kmot_vehicle.sh │ │ └── gen_labels_MOT.py │ │ └── visdrone │ │ └── visdrone2mot.py ├── picodet │ ├── FULL_QUANTIZATION.md │ ├── README.md │ ├── README_en.md │ ├── _base_ │ │ ├── optimizer_300e.yml │ │ ├── picodet_320_reader.yml │ │ ├── picodet_416_reader.yml │ │ ├── picodet_640_reader.yml │ │ └── picodet_v2.yml │ ├── application │ │ └── pedestrian_detection │ │ │ ├── picodet_s_192_lcnet_pedestrian.yml │ │ │ └── picodet_s_320_lcnet_pedestrian.yml │ ├── legacy_model │ │ ├── README.md │ │ ├── _base_ │ │ │ ├── optimizer_100e.yml │ │ │ ├── optimizer_300e.yml │ │ │ ├── picodet_320_reader.yml │ │ │ ├── picodet_416_reader.yml │ │ │ ├── picodet_640_reader.yml │ │ │ └── picodet_esnet.yml │ │ ├── application │ │ │ ├── layout_analysis │ │ │ │ ├── README.md │ │ │ │ ├── images │ │ │ │ │ ├── layout_demo.png │ │ │ │ │ └── layout_res.jpg │ │ │ │ ├── picodet_lcnet_x1_0_layout.yml │ │ │ │ └── picodet_lcnet_x2_5_layout.yml │ │ │ ├── mainbody_detection │ │ │ │ ├── README.md │ │ │ │ └── picodet_lcnet_x2_5_640_mainbody.yml │ │ │ └── pedestrian_detection │ │ │ │ ├── picodet_s_192_pedestrian.yml │ │ │ │ └── picodet_s_320_pedestrian.yml │ │ ├── more_config │ │ │ ├── picodet_lcnet_1_5x_416_coco.yml │ │ │ ├── picodet_lcnet_1_5x_640_coco.yml │ │ │ ├── picodet_mobilenetv3_large_1x_416_coco.yml │ │ │ ├── picodet_r18_640_coco.yml │ │ │ └── picodet_shufflenetv2_1x_416_coco.yml │ │ ├── picodet_l_320_coco.yml │ │ ├── picodet_l_416_coco.yml │ │ ├── picodet_l_640_coco.yml │ │ ├── picodet_m_320_coco.yml │ │ ├── picodet_m_416_coco.yml │ │ ├── picodet_s_320_coco.yml │ │ ├── picodet_s_320_voc.yml │ │ ├── picodet_s_416_coco.yml │ │ └── pruner │ │ │ ├── README.md │ │ │ ├── optimizer_300e_pruner.yml │ │ │ └── picodet_m_320_coco_pruner.yml │ ├── picodet_l_320_coco_lcnet.yml │ ├── picodet_l_416_coco_lcnet.yml │ ├── picodet_l_640_coco_lcnet.yml │ ├── picodet_m_320_coco_lcnet.yml │ ├── picodet_m_416_coco_lcnet.yml │ ├── picodet_s_320_coco_lcnet.yml │ ├── picodet_s_416_coco_lcnet.yml │ ├── picodet_s_416_coco_npu.yml │ ├── picodet_xs_320_coco_lcnet.yml │ └── picodet_xs_416_coco_lcnet.yml ├── pose3d │ ├── README.md │ ├── metro3d_24kpts.yml │ ├── tinypose3d_human36M.yml │ ├── tinypose3d_medical_multi_frames.yml │ └── tinypose3d_multi_frames_heatmap.yml ├── pphuman │ ├── README.md │ ├── dark_hrnet_w32_256x192.yml │ ├── pedestrian_yolov3 │ │ ├── README.md │ │ ├── README_cn.md │ │ ├── demo │ │ │ ├── 001.png │ │ │ ├── 002.png │ │ │ ├── 003.png │ │ │ └── 004.png │ │ ├── pedestrian.json │ │ └── pedestrian_yolov3_darknet.yml │ ├── ppyoloe_crn_l_36e_crowdhuman.yml │ ├── ppyoloe_crn_l_36e_pphuman.yml │ ├── ppyoloe_crn_s_36e_crowdhuman.yml │ ├── ppyoloe_crn_s_36e_pphuman.yml │ ├── ppyoloe_crn_s_80e_smoking_visdrone.yml │ └── ppyoloe_plus_crn_t_auxhead_320_60e_pphuman.yml ├── ppvehicle │ ├── README.md │ ├── mot_ppyoloe_l_36e_ppvehicle.yml │ ├── mot_ppyoloe_l_36e_ppvehicle9cls.yml │ ├── mot_ppyoloe_s_36e_ppvehicle.yml │ ├── mot_ppyoloe_s_36e_ppvehicle9cls.yml │ ├── ppyoloe_crn_l_36e_bdd100kdet.yml │ ├── ppyoloe_crn_l_36e_bdd100kmot.yml │ ├── ppyoloe_crn_l_36e_uadetrac.yml │ ├── ppyoloe_plus_crn_t_auxhead_320_60e_ppvehicle.yml │ └── vehicle_yolov3 │ │ ├── README.md │ │ ├── README_cn.md │ │ ├── demo │ │ ├── 001.jpeg │ │ ├── 003.png │ │ ├── 004.png │ │ └── 005.png │ │ ├── vehicle.json │ │ └── vehicle_yolov3_darknet.yml ├── ppyolo │ ├── README.md │ ├── README_cn.md │ ├── _base_ │ │ ├── optimizer_1x.yml │ │ ├── optimizer_2x.yml │ │ ├── optimizer_365e.yml │ │ ├── optimizer_650e.yml │ │ ├── ppyolo_mbv3_large.yml │ │ ├── ppyolo_mbv3_small.yml │ │ ├── ppyolo_r18vd.yml │ │ ├── ppyolo_r50vd_dcn.yml │ │ ├── ppyolo_reader.yml │ │ ├── ppyolo_tiny.yml │ │ ├── ppyolo_tiny_reader.yml │ │ ├── ppyolov2_r50vd_dcn.yml │ │ └── ppyolov2_reader.yml │ ├── ppyolo_mbv3_large_coco.yml │ ├── ppyolo_mbv3_small_coco.yml │ ├── ppyolo_r18vd_coco.yml │ ├── ppyolo_r50vd_dcn_1x_coco.yml │ ├── ppyolo_r50vd_dcn_1x_minicoco.yml │ ├── ppyolo_r50vd_dcn_2x_coco.yml │ ├── ppyolo_r50vd_dcn_voc.yml │ ├── ppyolo_test.yml │ ├── ppyolo_tiny_650e_coco.yml │ ├── ppyolov2_r101vd_dcn_365e_coco.yml │ ├── ppyolov2_r50vd_dcn_365e_coco.yml │ └── ppyolov2_r50vd_dcn_voc.yml ├── ppyoloe │ ├── README.md │ ├── README_cn.md │ ├── README_legacy.md │ ├── _base_ │ │ ├── optimizer_300e.yml │ │ ├── optimizer_36e_xpu.yml │ │ ├── optimizer_400e.yml │ │ ├── optimizer_60e.yml │ │ ├── optimizer_80e.yml │ │ ├── ppyoloe_crn.yml │ │ ├── ppyoloe_plus_crn.yml │ │ ├── ppyoloe_plus_crn_tiny_auxhead.yml │ │ ├── ppyoloe_plus_reader.yml │ │ ├── ppyoloe_plus_reader_320.yml │ │ └── ppyoloe_reader.yml │ ├── application │ │ ├── README.md │ │ ├── _base_ │ │ │ ├── exdark_detection.yml │ │ │ ├── pcb_detection.yml │ │ │ ├── sku110k.yml │ │ │ └── wgisd_detection.yml │ │ ├── ppyoloe_crn_m_80e_exdark.yml │ │ ├── ppyoloe_crn_m_80e_pcb.yml │ │ ├── ppyoloe_crn_m_80e_wgisd.yml │ │ ├── ppyoloe_plus_crn_l_80e_sku110k.yml │ │ ├── ppyoloe_plus_crn_m_80e_coco_pretrained_exdark.yml │ │ ├── ppyoloe_plus_crn_m_80e_coco_pretrained_pcb.yml │ │ ├── ppyoloe_plus_crn_m_80e_coco_pretrained_wgisd.yml │ │ ├── ppyoloe_plus_crn_m_80e_obj365_pretrained_exdark.yml │ │ ├── ppyoloe_plus_crn_m_80e_obj365_pretrained_pcb.yml │ │ ├── ppyoloe_plus_crn_m_80e_obj365_pretrained_wgisd.yml │ │ ├── ppyoloe_plus_crn_m_80e_sku110k.yml │ │ ├── ppyoloe_plus_crn_s_80e_sku110k.yml │ │ └── ppyoloe_plus_crn_x_80e_sku110k.yml │ ├── distill │ │ ├── README.md │ │ ├── ppyoloe_plus_crn_l_80e_coco_distill.yml │ │ ├── ppyoloe_plus_crn_m_80e_coco_distill.yml │ │ └── ppyoloe_plus_crn_s_80e_coco_distill.yml │ ├── objects365 │ │ ├── README_cn.md │ │ ├── ppyoloe_plus_crn_l_60e_objects365.yml │ │ ├── ppyoloe_plus_crn_m_60e_objects365.yml │ │ ├── ppyoloe_plus_crn_s_60e_objects365.yml │ │ └── ppyoloe_plus_crn_x_60e_objects365.yml │ ├── ppyoloe_crn_l_300e_coco.yml │ ├── ppyoloe_crn_l_36e_coco_xpu.yml │ ├── ppyoloe_crn_m_300e_coco.yml │ ├── ppyoloe_crn_s_300e_coco.yml │ ├── ppyoloe_crn_s_400e_coco.yml │ ├── ppyoloe_crn_x_300e_coco.yml │ ├── ppyoloe_plus_crn_l_30e_voc.yml │ ├── ppyoloe_plus_crn_l_80e_coco.yml │ ├── ppyoloe_plus_crn_m_80e_coco.yml │ ├── ppyoloe_plus_crn_s_30e_voc.yml │ ├── ppyoloe_plus_crn_s_80e_coco.yml │ ├── ppyoloe_plus_crn_t_auxhead_300e_coco.yml │ ├── ppyoloe_plus_crn_t_auxhead_320_300e_coco.yml │ ├── ppyoloe_plus_crn_t_auxhead_relu_300e_coco.yml │ ├── ppyoloe_plus_crn_t_auxhead_relu_320_300e_coco.yml │ ├── ppyoloe_plus_crn_x_80e_coco.yml │ └── voc │ │ ├── README_cn.md │ │ ├── ppyoloe_plus_crn_l_30e_voc.yml │ │ └── ppyoloe_plus_crn_s_30e_voc.yml ├── queryinst │ ├── README.md │ ├── _base_ │ │ ├── optimizer_1x.yml │ │ ├── queryinst_r50_fpn.yml │ │ └── queryinst_reader.yml │ ├── queryinst_r50_fpn_1x_pro100_coco.yml │ └── queryinst_r50_fpn_ms_crop_3x_pro300_coco.yml ├── rcnn_enhance │ ├── README.md │ ├── README_en.md │ ├── _base_ │ │ ├── faster_rcnn_enhance.yml │ │ ├── faster_rcnn_enhance_reader.yml │ │ └── optimizer_3x.yml │ └── faster_rcnn_enhance_3x_coco.yml ├── res2net │ ├── README.md │ ├── faster_rcnn_res2net50_vb_26w_4s_fpn_1x_coco.yml │ ├── mask_rcnn_res2net50_vb_26w_4s_fpn_2x_coco.yml │ └── mask_rcnn_res2net50_vd_26w_4s_fpn_2x_coco.yml ├── retinanet │ ├── README.md │ ├── _base_ │ │ ├── optimizer_1x.yml │ │ ├── optimizer_2x.yml │ │ ├── retinanet_r50_fpn.yml │ │ └── retinanet_reader.yml │ ├── retinanet_r101_distill_r50_2x_coco.yml │ ├── retinanet_r101_fpn_2x_coco.yml │ ├── retinanet_r50_fpn_1x_coco.yml │ └── retinanet_r50_fpn_2x_coco.yml ├── rotate │ ├── README.md │ ├── README_en.md │ ├── fcosr │ │ ├── README.md │ │ ├── README_en.md │ │ ├── _base_ │ │ │ ├── fcosr_reader.yml │ │ │ ├── fcosr_x50.yml │ │ │ └── optimizer_3x.yml │ │ └── fcosr_x50_3x_dota.yml │ ├── ppyoloe_r │ │ ├── README.md │ │ ├── README_en.md │ │ ├── _base_ │ │ │ ├── optimizer_3x.yml │ │ │ ├── ppyoloe_r_crn.yml │ │ │ └── ppyoloe_r_reader.yml │ │ ├── ppyoloe_r_crn_l_3x_dota.yml │ │ ├── ppyoloe_r_crn_l_3x_dota_ms.yml │ │ ├── ppyoloe_r_crn_m_3x_dota.yml │ │ ├── ppyoloe_r_crn_m_3x_dota_ms.yml │ │ ├── ppyoloe_r_crn_s_3x_dota.yml │ │ ├── ppyoloe_r_crn_s_3x_dota_ms.yml │ │ ├── ppyoloe_r_crn_x_3x_dota.yml │ │ └── ppyoloe_r_crn_x_3x_dota_ms.yml │ ├── s2anet │ │ ├── README.md │ │ ├── README_en.md │ │ ├── _base_ │ │ │ ├── s2anet.yml │ │ │ ├── s2anet_optimizer_1x.yml │ │ │ ├── s2anet_optimizer_2x.yml │ │ │ └── s2anet_reader.yml │ │ ├── s2anet_1x_spine.yml │ │ ├── s2anet_alignconv_2x_dota.yml │ │ └── s2anet_conv_2x_dota.yml │ └── tools │ │ ├── convert.py │ │ ├── generate_result.py │ │ ├── inference_benchmark.py │ │ ├── onnx_infer.py │ │ ├── prepare_data.py │ │ └── slicebase.py ├── rtdetr │ ├── README.md │ ├── _base_ │ │ ├── optimizer_6x.yml │ │ ├── rtdetr_r50vd.yml │ │ └── rtdetr_reader.yml │ ├── rtdetr_hgnetv2_l_6x_coco.yml │ ├── rtdetr_hgnetv2_x_6x_coco.yml │ ├── rtdetr_r101vd_6x_coco.yml │ ├── rtdetr_r18vd_6x_coco.yml │ ├── rtdetr_r34vd_6x_coco.yml │ ├── rtdetr_r50vd_6x_coco.yml │ └── rtdetr_r50vd_m_6x_coco.yml ├── rtmdet │ ├── README.md │ ├── _base_ │ │ ├── optimizer_300e.yml │ │ ├── rtmdet_cspnext.yml │ │ └── rtmdet_reader.yml │ ├── rtmdet_l_300e_coco.yml │ ├── rtmdet_m_300e_coco.yml │ ├── rtmdet_s_300e_coco.yml │ ├── rtmdet_t_300e_coco.yml │ └── rtmdet_x_300e_coco.yml ├── runtime.yml ├── semi_det │ ├── README.md │ ├── _base_ │ │ ├── coco_detection_full.yml │ │ ├── coco_detection_percent_1.yml │ │ ├── coco_detection_percent_10.yml │ │ ├── coco_detection_percent_5.yml │ │ ├── coco_detection_voc.yml │ │ └── voc2coco.py │ ├── arsl │ │ ├── README.md │ │ ├── _base_ │ │ │ ├── arsl_fcos_r50_fpn.yml │ │ │ ├── arsl_fcos_reader.yml │ │ │ ├── optimizer_360k.yml │ │ │ └── optimizer_90k.yml │ │ ├── arsl_fcos_r50_fpn_coco_full.yml │ │ ├── arsl_fcos_r50_fpn_coco_semi001.yml │ │ ├── arsl_fcos_r50_fpn_coco_semi005.yml │ │ ├── arsl_fcos_r50_fpn_coco_semi010.yml │ │ └── arsl_fcos_r50_fpn_coco_semi010_lsj.yml │ ├── baseline │ │ ├── README.md │ │ ├── faster_rcnn_r50_fpn_2x_coco_sup005.yml │ │ ├── faster_rcnn_r50_fpn_2x_coco_sup010.yml │ │ ├── fcos_r50_fpn_2x_coco_sup005.yml │ │ ├── fcos_r50_fpn_2x_coco_sup010.yml │ │ ├── ppyoloe_plus_crn_l_80e_coco_sup005.yml │ │ ├── ppyoloe_plus_crn_l_80e_coco_sup010.yml │ │ ├── ppyoloe_plus_crn_s_80e_coco_sup005.yml │ │ ├── ppyoloe_plus_crn_s_80e_coco_sup010.yml │ │ ├── retinanet_r50_fpn_2x_coco_sup005.yml │ │ └── retinanet_r50_fpn_2x_coco_sup010.yml │ └── denseteacher │ │ ├── README.md │ │ ├── denseteacher_fcos_r50_fpn_coco_full.yml │ │ ├── denseteacher_fcos_r50_fpn_coco_semi005.yml │ │ ├── denseteacher_fcos_r50_fpn_coco_semi010.yml │ │ ├── denseteacher_fcos_r50_fpn_coco_semi010_lsj.yml │ │ ├── denseteacher_ppyoloe_plus_crn_l_coco_semi005.yml │ │ ├── denseteacher_ppyoloe_plus_crn_l_coco_semi010.yml │ │ ├── denseteacher_ppyoloe_plus_crn_s_coco_semi005.yml │ │ └── denseteacher_ppyoloe_plus_crn_s_coco_semi010.yml ├── slim │ ├── README.md │ ├── README_en.md │ ├── distill │ │ ├── README.md │ │ ├── gfl_ld_distill.yml │ │ ├── gfl_r101vd_fpn_coco_distill_cwd.yml │ │ ├── ppyoloe_plus_distill_l_distill_m.yml │ │ ├── ppyoloe_plus_distill_m_distill_s.yml │ │ ├── ppyoloe_plus_distill_x_distill_l.yml │ │ ├── retinanet_resnet101_coco_distill.yml │ │ ├── retinanet_resnet101_coco_distill_cwd.yml │ │ └── yolov3_mobilenet_v1_coco_distill.yml │ ├── extensions │ │ ├── yolov3_mobilenet_v1_coco_distill_prune.yml │ │ └── yolov3_mobilenetv1_prune_qat.yml │ ├── ofa │ │ └── ofa_picodet_demo.yml │ ├── post_quant │ │ ├── mask_rcnn_r50_fpn_1x_coco_ptq.yml │ │ ├── mcfairmot_ptq.yml │ │ ├── picodet_s_ptq.yml │ │ ├── ppyolo_mbv3_large_ptq.yml │ │ ├── ppyolo_r50vd_dcn_ptq.yml │ │ ├── ppyoloe_crn_s_300e_coco_ptq.yml │ │ ├── tinypose_128x96_ptq.yml │ │ └── yolov3_darknet53_ptq.yml │ ├── prune │ │ ├── faster_rcnn_r50_fpn_prune_fpgm.yml │ │ ├── picodet_m_unstructured_prune_75.yml │ │ ├── picodet_m_unstructured_prune_85.yml │ │ ├── ppyolo_mbv3_large_prune_fpgm.yml │ │ ├── ppyolo_r50vd_prune_fpgm.yml │ │ ├── yolov3_darknet_prune_fpgm.yml │ │ ├── yolov3_prune_fpgm.yml │ │ └── yolov3_prune_l1_norm.yml │ └── quant │ │ ├── mask_rcnn_r50_fpn_1x_qat.yml │ │ ├── picodet_s_416_lcnet_quant.yml │ │ ├── picodet_s_quant.yml │ │ ├── ppyolo_mbv3_large_qat.yml │ │ ├── ppyolo_r50vd_qat_pact.yml │ │ ├── ppyoloe_l_qat.yml │ │ ├── ppyolov2_r50vd_dcn_qat.yml │ │ ├── ssd_mobilenet_v1_qat.yml │ │ ├── tinypose_qat.yml │ │ ├── yolov3_darknet_qat.yml │ │ ├── yolov3_mobilenet_v1_qat.yml │ │ └── yolov3_mobilenet_v3_qat.yml ├── smalldet │ ├── DataDownload.md │ ├── README.md │ ├── _base_ │ │ ├── DOTA_sliced_500_025_detection.yml │ │ ├── visdrone_sliced_640_025_detection.yml │ │ └── xview_sliced_400_025_detection.yml │ ├── ppyoloe_crn_l_80e_sliced_visdrone_640_025.yml │ ├── ppyoloe_crn_l_80e_sliced_visdrone_640_025_slice_infer.yml │ ├── ppyoloe_p2_crn_l_80e_sliced_DOTA_500_025.yml │ ├── ppyoloe_p2_crn_l_80e_sliced_xview_400_025.yml │ ├── ppyoloe_plus_sod_crn_l_80e_coco.yml │ └── visdrone │ │ ├── README.md │ │ ├── ppyoloe_crn_l_80e_visdrone.yml │ │ ├── ppyoloe_crn_l_alpha_largesize_80e_visdrone.yml │ │ ├── ppyoloe_crn_l_p2_alpha_80e_visdrone.yml │ │ ├── ppyoloe_crn_l_p2_alpha_largesize_80e_visdrone.yml │ │ ├── ppyoloe_crn_s_80e_visdrone.yml │ │ ├── ppyoloe_crn_s_p2_alpha_80e_visdrone.yml │ │ ├── ppyoloe_plus_crn_l_largesize_80e_visdrone.yml │ │ ├── ppyoloe_plus_sod_crn_l_80e_visdrone.yml │ │ ├── ppyoloe_plus_sod_crn_l_largesize_80e_visdrone.yml │ │ └── ppyoloe_plus_sod_crn_s_80e_visdrone.yml ├── smrt │ ├── DataAnalysis.md │ ├── README.md │ ├── images │ │ ├── 00362.jpg │ │ ├── Board_diaojiao_1591.png │ │ ├── UpCoa_liewen_163.png │ │ ├── lvjian1_0.jpg │ │ ├── lvjian1_10.jpg │ │ ├── renche_00002.jpg │ │ └── renche_00204.jpg │ ├── picodet │ │ ├── picodet_l_1024_coco_lcnet_battery.yml │ │ ├── picodet_l_1024_coco_lcnet_lvjian1.yml │ │ ├── picodet_l_1024_coco_lcnet_renche.yml │ │ ├── picodet_l_640_coco_lcnet_battery.yml │ │ ├── picodet_l_640_coco_lcnet_lvjian1.yml │ │ └── picodet_l_640_coco_lcnet_renche.yml │ ├── ppyolo │ │ ├── ppyolov2_r101vd_dcn_365e_battery.yml │ │ ├── ppyolov2_r101vd_dcn_365e_battery_1024.yml │ │ ├── ppyolov2_r101vd_dcn_365e_lvjian1_1024.yml │ │ ├── ppyolov2_r101vd_dcn_365e_lvjian1_640.yml │ │ ├── ppyolov2_r101vd_dcn_365e_renche_1024.yml │ │ ├── ppyolov2_r101vd_dcn_365e_renche_640.yml │ │ ├── ppyolov2_r50vd_dcn_365e_battery.yml │ │ ├── ppyolov2_r50vd_dcn_365e_battery_1024.yml │ │ ├── ppyolov2_r50vd_dcn_365e_lvjian1_1024.yml │ │ ├── ppyolov2_r50vd_dcn_365e_lvjian1_640.yml │ │ ├── ppyolov2_r50vd_dcn_365e_renche_1024.yml │ │ └── ppyolov2_r50vd_dcn_365e_renche_640.yml │ ├── ppyoloe │ │ ├── ppyoloe_crn_l_300e_battery.yml │ │ ├── ppyoloe_crn_l_300e_battery_1024.yml │ │ ├── ppyoloe_crn_l_300e_lvjian1.yml │ │ ├── ppyoloe_crn_l_300e_lvjian1_1024.yml │ │ ├── ppyoloe_crn_l_300e_renche.yml │ │ ├── ppyoloe_crn_l_300e_renche_1024.yml │ │ ├── ppyoloe_crn_m_300e_battery.yml │ │ ├── ppyoloe_crn_m_300e_battery_1024.yml │ │ ├── ppyoloe_crn_m_300e_lvjian1.yml │ │ ├── ppyoloe_crn_m_300e_lvjian1_1024.yml │ │ ├── ppyoloe_crn_m_300e_renche.yml │ │ ├── ppyoloe_crn_m_300e_renche_1024.yml │ │ ├── ppyoloe_crn_s_300e_battery.yml │ │ ├── ppyoloe_crn_s_300e_battery_1024.yml │ │ ├── ppyoloe_crn_s_300e_lvjian1.yml │ │ ├── ppyoloe_crn_s_300e_lvjian_1024.yml │ │ ├── ppyoloe_crn_s_300e_renche.yml │ │ ├── ppyoloe_crn_s_300e_renche_1024.yml │ │ ├── ppyoloe_crn_x_300e_battery.yml │ │ ├── ppyoloe_crn_x_300e_battery_1024.yml │ │ ├── ppyoloe_crn_x_300e_lvjian1.yml │ │ ├── ppyoloe_crn_x_300e_lvjian1_1024.yml │ │ ├── ppyoloe_crn_x_300e_renche.yml │ │ └── ppyoloe_crn_x_300e_renche_1024.yml │ └── rcnn │ │ ├── cascade_rcnn_r50_vd_fpn_ssld_2x_1500_battery.yml │ │ ├── cascade_rcnn_r50_vd_fpn_ssld_2x_1500_lvjian1.yml │ │ ├── cascade_rcnn_r50_vd_fpn_ssld_2x_1500_renche.yml │ │ ├── cascade_rcnn_r50_vd_fpn_ssld_2x_800_battery.yml │ │ ├── cascade_rcnn_r50_vd_fpn_ssld_2x_800_lvjian1.yml │ │ ├── cascade_rcnn_r50_vd_fpn_ssld_2x_800_renche.yml │ │ ├── faster_rcnn_r101_vd_fpn_ssld_2x_1500_battery.yml │ │ ├── faster_rcnn_r101_vd_fpn_ssld_2x_1500_lvjian1.yml │ │ ├── faster_rcnn_r101_vd_fpn_ssld_2x_1500_renche.yml │ │ ├── faster_rcnn_r101_vd_fpn_ssld_2x_800_battery.yml │ │ ├── faster_rcnn_r101_vd_fpn_ssld_2x_800_lvjian1.yml │ │ ├── faster_rcnn_r101_vd_fpn_ssld_2x_800_renche.yml │ │ ├── faster_rcnn_r50_vd_fpn_ssld_2x_1500_battery.yml │ │ ├── faster_rcnn_r50_vd_fpn_ssld_2x_1500_lvjian1.yml │ │ ├── faster_rcnn_r50_vd_fpn_ssld_2x_1500_renche.yml │ │ ├── faster_rcnn_r50_vd_fpn_ssld_2x_800_battery.yml │ │ ├── faster_rcnn_r50_vd_fpn_ssld_2x_800_lvjian1.yml │ │ └── faster_rcnn_r50_vd_fpn_ssld_2x_800_renche.yml ├── sniper │ ├── README.md │ ├── README_cn.md │ ├── _base_ │ │ ├── faster_fpn_reader.yml │ │ ├── faster_reader.yml │ │ └── ppyolo_reader.yml │ ├── faster_rcnn_r50_fpn_1x_sniper_visdrone.yml │ ├── faster_rcnn_r50_fpn_1x_visdrone.yml │ ├── ppyolo_r50vd_dcn_1x_sniper_visdrone.yml │ └── ppyolo_r50vd_dcn_1x_visdrone.yml ├── solov2 │ ├── README.md │ ├── _base_ │ │ ├── optimizer_1x.yml │ │ ├── solov2_light_reader.yml │ │ ├── solov2_r50_fpn.yml │ │ └── solov2_reader.yml │ ├── solov2_r101_vd_fpn_3x_coco.yml │ ├── solov2_r50_enhance_coco.yml │ ├── solov2_r50_fpn_1x_coco.yml │ └── solov2_r50_fpn_3x_coco.yml ├── sparse_rcnn │ ├── README.md │ ├── _base_ │ │ ├── optimizer_3x.yml │ │ ├── sparse_rcnn_r50_fpn.yml │ │ └── sparse_rcnn_reader.yml │ ├── sparse_rcnn_r50_fpn_3x_pro100_coco.yml │ └── sparse_rcnn_r50_fpn_3x_pro300_coco.yml ├── ssd │ ├── README.md │ ├── _base_ │ │ ├── optimizer_120e.yml │ │ ├── optimizer_1700e.yml │ │ ├── optimizer_240e.yml │ │ ├── optimizer_70e.yml │ │ ├── ssd_mobilenet_reader.yml │ │ ├── ssd_mobilenet_v1_300.yml │ │ ├── ssd_r34_300.yml │ │ ├── ssd_r34_reader.yml │ │ ├── ssd_reader.yml │ │ ├── ssd_vgg16_300.yml │ │ ├── ssdlite300_reader.yml │ │ ├── ssdlite320_reader.yml │ │ ├── ssdlite_ghostnet_320.yml │ │ ├── ssdlite_mobilenet_v1_300.yml │ │ ├── ssdlite_mobilenet_v3_large_320.yml │ │ └── ssdlite_mobilenet_v3_small_320.yml │ ├── ssd_mobilenet_v1_300_120e_voc.yml │ ├── ssd_r34_70e_coco.yml │ ├── ssd_vgg16_300_240e_voc.yml │ ├── ssdlite_ghostnet_320_coco.yml │ ├── ssdlite_mobilenet_v1_300_coco.yml │ ├── ssdlite_mobilenet_v3_large_320_coco.yml │ └── ssdlite_mobilenet_v3_small_320_coco.yml ├── swin │ ├── README.md │ ├── faster_rcnn_swin_tiny_fpn_3x_coco.yml │ └── ppyoloe_plus_swin_tiny_36e_coco.yml ├── tood │ ├── README.md │ ├── _base_ │ │ ├── optimizer_1x.yml │ │ ├── tood_r50_fpn.yml │ │ └── tood_reader.yml │ └── tood_r50_fpn_1x_coco.yml ├── ttfnet │ ├── README.md │ ├── README_en.md │ ├── _base_ │ │ ├── optimizer_10x.yml │ │ ├── optimizer_1x.yml │ │ ├── optimizer_20x.yml │ │ ├── pafnet.yml │ │ ├── pafnet_lite.yml │ │ ├── pafnet_lite_reader.yml │ │ ├── pafnet_reader.yml │ │ ├── ttfnet_darknet53.yml │ │ └── ttfnet_reader.yml │ ├── pafnet_10x_coco.yml │ ├── pafnet_lite_mobilenet_v3_20x_coco.yml │ └── ttfnet_darknet53_1x_coco.yml ├── vitdet │ ├── README.md │ ├── _base_ │ │ ├── faster_rcnn_reader.yml │ │ ├── mask_rcnn_reader.yml │ │ ├── optimizer_base_1x.yml │ │ ├── optimizer_base_30e.yml │ │ ├── optimizer_base_36e.yml │ │ └── ppyoloe_reader.yml │ ├── cascade_rcnn_vit_base_hrfpn_cae_1x_coco.yml │ ├── cascade_rcnn_vit_large_hrfpn_cae_1x_coco.yml │ ├── faster_rcnn_vit_base_fpn_cae_1x_coco.yml │ ├── mask_rcnn_vit_base_hrfpn_cae_1x_coco.yml │ ├── mask_rcnn_vit_large_hrfpn_cae_1x_coco.yml │ ├── ppyoloe_vit_base_csppan_cae_30e_coco.yml │ └── ppyoloe_vit_base_csppan_cae_36e_coco.yml ├── yolof │ ├── README.md │ ├── _base_ │ │ ├── optimizer_1x.yml │ │ ├── yolof_r50_c5.yml │ │ └── yolof_reader.yml │ └── yolof_r50_c5_1x_coco.yml ├── yolov3 │ ├── README.md │ ├── _base_ │ │ ├── optimizer_270e.yml │ │ ├── optimizer_40e.yml │ │ ├── yolov3_darknet53.yml │ │ ├── yolov3_mobilenet_v1.yml │ │ ├── yolov3_mobilenet_v3_large.yml │ │ ├── yolov3_mobilenet_v3_small.yml │ │ ├── yolov3_r34.yml │ │ ├── yolov3_r50vd_dcn.yml │ │ └── yolov3_reader.yml │ ├── yolov3_darknet53_270e_coco.yml │ ├── yolov3_darknet53_270e_voc.yml │ ├── yolov3_darknet53_original_270e_coco.yml │ ├── yolov3_darknet53_original_320e_coco_1p.yml │ ├── yolov3_mobilenet_v1_270e_coco.yml │ ├── yolov3_mobilenet_v1_270e_voc.yml │ ├── yolov3_mobilenet_v1_roadsign.yml │ ├── yolov3_mobilenet_v1_ssld_270e_coco.yml │ ├── yolov3_mobilenet_v1_ssld_270e_voc.yml │ ├── yolov3_mobilenet_v3_large_270e_coco.yml │ ├── yolov3_mobilenet_v3_large_270e_voc.yml │ ├── yolov3_mobilenet_v3_large_ssld_270e_voc.yml │ ├── yolov3_r34_270e_coco.yml │ └── yolov3_r50vd_dcn_270e_coco.yml ├── yolov5 │ ├── README.md │ ├── _base_ │ │ ├── optimizer_300e.yml │ │ ├── optimizer_300e_high.yml │ │ ├── yolov5_cspdarknet.yml │ │ ├── yolov5_cspresnet.yml │ │ ├── yolov5_reader.yml │ │ ├── yolov5_reader_high_aug.yml │ │ ├── yolov5p2_cspdarknet.yml │ │ ├── yolov5p6_cspdarknet.yml │ │ └── yolov5p6_reader.yml │ ├── yolov5_l_300e_coco.yml │ ├── yolov5_m_300e_coco.yml │ ├── yolov5_n_300e_coco.yml │ ├── yolov5_s_300e_coco.yml │ ├── yolov5_x_300e_coco.yml │ ├── yolov5p6_l_300e_coco.yml │ ├── yolov5p6_m_300e_coco.yml │ ├── yolov5p6_n_300e_coco.yml │ ├── yolov5p6_s_300e_coco.yml │ └── yolov5p6_x_300e_coco.yml ├── yolov5u │ ├── README.md │ ├── _base_ │ │ ├── optimizer_300e.yml │ │ ├── optimizer_300e_high.yml │ │ ├── yolov5u_cspdarknet.yml │ │ ├── yolov5u_reader.yml │ │ ├── yolov5u_reader_high_aug.yml │ │ ├── yolov5up6_cspdarknet.yml │ │ └── yolov5up6_reader.yml │ ├── yolov5u_l_300e_coco.yml │ ├── yolov5u_m_300e_coco.yml │ ├── yolov5u_n_300e_coco.yml │ ├── yolov5u_s_300e_coco.yml │ ├── yolov5u_x_300e_coco.yml │ ├── yolov5up6_l_300e_coco.yml │ ├── yolov5up6_m_300e_coco.yml │ ├── yolov5up6_n_300e_coco.yml │ ├── yolov5up6_s_300e_coco.yml │ └── yolov5up6_x_300e_coco.yml ├── yolov6 │ ├── README.md │ ├── _base_ │ │ ├── optimizer_300e.yml │ │ ├── optimizer_300e_fintune.yml │ │ ├── yolov6_cspbep.yml │ │ ├── yolov6_efficientrep.yml │ │ ├── yolov6_reader.yml │ │ ├── yolov6_reader_fintune.yml │ │ └── yolov6_reader_high_aug.yml │ ├── yolov6_l_300e_coco.yml │ ├── yolov6_m_300e_coco.yml │ ├── yolov6_n_300e_coco.yml │ ├── yolov6_s_300e_coco.yml │ └── yolov6lite │ │ ├── README.md │ │ ├── _base_ │ │ ├── optimizer_400e.yml │ │ ├── optimizer_400e_fintune.yml │ │ ├── yolov6lite_effibackbone.yml │ │ └── yolov6lite_reader_320.yml │ │ ├── yolov6lite_l_400e_coco.yml │ │ ├── yolov6lite_m_400e_coco.yml │ │ └── yolov6lite_s_400e_coco.yml ├── yolov7 │ ├── README.md │ ├── _base_ │ │ ├── optimizer_300e.yml │ │ ├── optimizer_300e_p6.yml │ │ ├── optimizer_300e_tiny.yml │ │ ├── yolov7_elannet.yml │ │ ├── yolov7_reader.yml │ │ ├── yolov7_tiny_reader.yml │ │ ├── yolov7p6_elannet.yml │ │ └── yolov7p6_reader.yml │ ├── yolov7_l_300e_coco.yml │ ├── yolov7_tiny_300e_coco.yml │ ├── yolov7_tiny_320_300e_coco.yml │ ├── yolov7_tiny_416_300e_coco.yml │ ├── yolov7_x_300e_coco.yml │ ├── yolov7p6_d6_300e_coco.yml │ ├── yolov7p6_e6_300e_coco.yml │ ├── yolov7p6_e6e_300e_coco.yml │ └── yolov7p6_w6_300e_coco.yml ├── yolov7u │ ├── README.md │ ├── _base_ │ │ ├── optimizer_300e.yml │ │ ├── yolov7u_elannet.yml │ │ └── yolov7u_reader.yml │ └── yolov7u_l_300e_coco.yml ├── yolov8 │ ├── README.md │ ├── _base_ │ │ ├── optimizer_100e.yml │ │ ├── optimizer_300e.yml │ │ ├── optimizer_500e.yml │ │ ├── optimizer_500e_high.yml │ │ ├── yolov8_cspdarknet.yml │ │ ├── yolov8_reader.yml │ │ ├── yolov8_reader_high_aug.yml │ │ ├── yolov8p6_cspdarknet.yml │ │ └── yolov8p6_reader.yml │ ├── yolov8_l_500e_coco.yml │ ├── yolov8_m_500e_coco.yml │ ├── yolov8_n_500e_coco.yml │ ├── yolov8_s_500e_coco.yml │ ├── yolov8_x_500e_coco.yml │ └── yolov8p6_x_500e_coco.yml └── yolox │ ├── README.md │ ├── _base_ │ ├── optimizer_300e.yml │ ├── yolox_cspdarknet.yml │ └── yolox_reader.yml │ ├── yolox_cdn_tiny_300e_coco.yml │ ├── yolox_crn_s_300e_coco.yml │ ├── yolox_l_300e_coco.yml │ ├── yolox_m_300e_coco.yml │ ├── yolox_nano_300e_coco.yml │ ├── yolox_s_300e_coco.yml │ ├── yolox_tiny_300e_coco.yml │ └── yolox_x_300e_coco.yml ├── demo ├── 000000014439.jpg ├── 000000014439_640x640.jpg ├── 000000087038.jpg ├── 000000570688.jpg ├── 0000315_01601_d_0000509.jpg ├── 39006.jpg ├── P0072__1.0__0___0.png ├── P0861__1.0__1154___824.png ├── car.jpg ├── hrnet_demo.jpg ├── orange_71.jpg └── road554.png ├── docs ├── CHANGELOG.md ├── CHANGELOG_en.md ├── MODEL_ZOO_cn.md ├── MODEL_ZOO_en.md ├── advanced_tutorials │ ├── MODEL_TECHNICAL.md │ ├── MODEL_TECHNICAL_en.md │ ├── READER.md │ ├── READER_en.md │ ├── customization │ │ ├── action_recognotion │ │ │ ├── README.md │ │ │ ├── README_en.md │ │ │ ├── idbased_clas.md │ │ │ ├── idbased_clas_en.md │ │ │ ├── idbased_det.md │ │ │ ├── idbased_det_en.md │ │ │ ├── skeletonbased_rec.md │ │ │ ├── skeletonbased_rec_en.md │ │ │ └── videobased_rec.md │ │ ├── detection.md │ │ ├── detection_en.md │ │ ├── keypoint_detection.md │ │ ├── keypoint_detection_en.md │ │ ├── pphuman_attribute.md │ │ ├── pphuman_attribute_en.md │ │ ├── pphuman_mot.md │ │ ├── pphuman_mot_en.md │ │ ├── pphuman_mtmct.md │ │ ├── pphuman_mtmct_en.md │ │ ├── ppvehicle_attribute.md │ │ ├── ppvehicle_attribute_en.md │ │ ├── ppvehicle_plate.md │ │ ├── ppvehicle_violation.md │ │ └── ppvehicle_violation_en.md │ └── openvino_inference │ │ ├── README.md │ │ ├── README_cn.md │ │ ├── fairmot_onnx_openvino.py │ │ ├── requirements.txt │ │ ├── street.jpeg │ │ └── tracked.jpg ├── contribution │ ├── README.md │ └── Yes_PP-YOLOE.md ├── feature_models │ ├── PaddleYOLO_MODEL.md │ ├── PaddleYOLO_MODEL_en.md │ ├── SSLD_PRETRAINED_MODEL.md │ └── SSLD_PRETRAINED_MODEL_en.md ├── images │ ├── 000000014439.jpg │ ├── 12_Group_Group_12_Group_Group_12_935.jpg │ ├── PedestrianDetection_001.png │ ├── PedestrianDetection_004.png │ ├── VehicleDetection_001.jpeg │ ├── VehicleDetection_005.png │ ├── add_attribute.png │ ├── bus.jpg │ ├── dog.jpg │ ├── fps_map.png │ ├── grad_cam_ppyoloe_demo.jpg │ ├── input_shape.png │ ├── instance_seg.png │ ├── layout.jpg │ ├── layout_demo1.png │ ├── layout_demo2.png │ ├── lite_demo.jpg │ ├── mobile_fps_map.png │ ├── model_figure.png │ ├── picedet_demo.jpeg │ ├── picodet_android_demo1.jpg │ ├── picodet_android_demo2.jpg │ ├── picodet_android_demo3.jpg │ ├── picodet_android_demo4.jpg │ ├── picodet_map.png │ ├── pphuman-tech.png │ ├── pphumanv2.png │ ├── pptracking.png │ ├── pptracking_en.png │ ├── ppvehicle.png │ ├── ppyolo_map_fps.png │ ├── ppyoloe_plus_map_fps.png │ ├── ppyoloe_r_map_fps.png │ ├── reader_figure.png │ ├── res.jpg │ ├── road554.png │ ├── roadsign_yml.png │ ├── ssld_model.png │ ├── tinypose_app.png │ ├── tinypose_demo.png │ ├── tinypose_pipeline.png │ └── yaml_show.png └── tutorials │ ├── DistributedTraining_cn.md │ ├── DistributedTraining_en.md │ ├── FAQ │ ├── FAQ第一期.md │ ├── FAQ第零期.md │ └── README.md │ ├── GETTING_STARTED.md │ ├── GETTING_STARTED_cn.md │ ├── GradCAM_cn.md │ ├── GradCAM_en.md │ ├── INSTALL.md │ ├── INSTALL_cn.md │ ├── KeyPointConfigGuide_cn.md │ ├── KeyPointConfigGuide_en.md │ ├── QUICK_STARTED.md │ ├── QUICK_STARTED_cn.md │ ├── config_annotation │ ├── faster_rcnn_r50_fpn_1x_coco_annotation.md │ ├── faster_rcnn_r50_fpn_1x_coco_annotation_en.md │ ├── multi_scale_test_config.md │ ├── multi_scale_test_config_cn.md │ ├── ppyolo_r50vd_dcn_1x_coco_annotation.md │ └── ppyolo_r50vd_dcn_1x_coco_annotation_en.md │ ├── data │ ├── DetAnnoTools.md │ ├── DetAnnoTools_en.md │ ├── KeyPointAnnoTools.md │ ├── KeyPointAnnoTools_en.md │ ├── MOTAnnoTools.md │ ├── PrepareDetDataSet.md │ ├── PrepareDetDataSet_en.md │ ├── PrepareKeypointDataSet.md │ ├── PrepareKeypointDataSet_en.md │ ├── PrepareMOTDataSet.md │ ├── PrepareMOTDataSet_en.md │ └── README.md │ └── logging_en.md ├── ppdettorch ├── __init__.py ├── core │ ├── __init__.py │ ├── config │ │ ├── __init__.py │ │ ├── schema.py │ │ └── yaml_helpers.py │ └── workspace.py ├── data │ ├── __init__.py │ ├── crop_utils │ │ ├── __init__.py │ │ ├── annotation_cropper.py │ │ └── chip_box_utils.py │ ├── reader.py │ ├── shm_utils.py │ ├── source │ │ ├── __init__.py │ │ ├── category.py │ │ ├── coco.py │ │ ├── dataset.py │ │ ├── keypoint_coco.py │ │ ├── mot.py │ │ ├── sniper_coco.py │ │ ├── voc.py │ │ └── widerface.py │ ├── transform │ │ ├── __init__.py │ │ ├── atss_assigner.py │ │ ├── autoaugment_utils.py │ │ ├── batch_operators.py │ │ ├── gridmask_utils.py │ │ ├── keypoint_operators.py │ │ ├── mot_operators.py │ │ ├── op_helper.py │ │ ├── operators.py │ │ └── rotated_operators.py │ └── utils.py ├── engine │ ├── __init__.py │ └── detection_trainer.py ├── metrics │ ├── __init__.py │ ├── coco_utils.py │ ├── json_results.py │ ├── keypoint_metrics.py │ ├── map_utils.py │ ├── mcmot_metrics.py │ ├── metrics.py │ ├── mot_metrics.py │ ├── munkres.py │ └── widerface_utils.py ├── modeling │ ├── __init__.py │ ├── architectures │ │ ├── __init__.py │ │ ├── blazeface.py │ │ ├── bytetrack.py │ │ ├── cascade_rcnn.py │ │ ├── centernet.py │ │ ├── deepsort.py │ │ ├── detr.py │ │ ├── fairmot.py │ │ ├── faster_rcnn.py │ │ ├── fcos.py │ │ ├── gfl.py │ │ ├── jde.py │ │ ├── keypoint_hrhrnet.py │ │ ├── keypoint_hrnet.py │ │ ├── mask_rcnn.py │ │ ├── meta_arch.py │ │ ├── picodet.py │ │ ├── ppyoloe.py │ │ ├── retinanet.py │ │ ├── rtmdet.py │ │ ├── s2anet.py │ │ ├── solov2.py │ │ ├── sparse_rcnn.py │ │ ├── ssd.py │ │ ├── tood.py │ │ ├── ttfnet.py │ │ ├── yolo.py │ │ ├── yolov5.py │ │ ├── yolov6.py │ │ ├── yolov7.py │ │ ├── yolov8.py │ │ └── yolox.py │ ├── assigners │ │ ├── __init__.py │ │ ├── atss_assigner.py │ │ ├── max_iou_assigner.py │ │ ├── simota_assigner.py │ │ ├── task_aligned_assigner.py │ │ └── utils.py │ ├── backbones │ │ ├── __init__.py │ │ ├── blazenet.py │ │ ├── convnext.py │ │ ├── csp_darknet.py │ │ ├── cspnext.py │ │ ├── cspresnet.py │ │ ├── darknet.py │ │ ├── dla.py │ │ ├── esnet.py │ │ ├── ghostnet.py │ │ ├── hardnet.py │ │ ├── hgnet_v2.py │ │ ├── hrnet.py │ │ ├── lcnet.py │ │ ├── lite_hrnet.py │ │ ├── mobilenet_v1.py │ │ ├── mobilenet_v3.py │ │ ├── mobileone.py │ │ ├── name_adapter.py │ │ ├── res2net.py │ │ ├── resnet.py │ │ ├── senet.py │ │ ├── shufflenet_v2.py │ │ ├── swin_transformer.py │ │ ├── transformer_utils.py │ │ ├── vgg.py │ │ ├── vision_transformer.py │ │ ├── yolov6_efficientrep.py │ │ ├── yolov7_elannet.py │ │ └── yolov8_csp_darknet.py │ ├── bbox_utils.py │ ├── cls_utils.py │ ├── heads │ │ ├── __init__.py │ │ ├── bbox_head.py │ │ ├── cascade_head.py │ │ ├── centernet_head.py │ │ ├── detr_head.py │ │ ├── face_head.py │ │ ├── fcos_head.py │ │ ├── gfl_head.py │ │ ├── keypoint_hrhrnet_head.py │ │ ├── mask_head.py │ │ ├── pico_head.py │ │ ├── ppyoloe_head.py │ │ ├── ppyoloe_r_head.py │ │ ├── retina_head.py │ │ ├── roi_extractor.py │ │ ├── rtmdet_head.py │ │ ├── s2anet_head.py │ │ ├── simota_head.py │ │ ├── solov2_head.py │ │ ├── sparsercnn_head.py │ │ ├── ssd_head.py │ │ ├── tood_head.py │ │ ├── ttf_head.py │ │ ├── yolo_head.py │ │ ├── yolov5_head.py │ │ ├── yolov6_head.py │ │ ├── yolov7_head.py │ │ └── yolov8_head.py │ ├── keypoint_utils.py │ ├── layers.py │ ├── losses │ │ ├── __init__.py │ │ ├── ctfocal_loss.py │ │ ├── detr_loss.py │ │ ├── fairmot_loss.py │ │ ├── fcos_loss.py │ │ ├── focal_loss.py │ │ ├── gfocal_loss.py │ │ ├── iou_aware_loss.py │ │ ├── iou_loss.py │ │ ├── jde_loss.py │ │ ├── keypoint_loss.py │ │ ├── probiou_loss.py │ │ ├── smooth_l1_loss.py │ │ ├── solov2_loss.py │ │ ├── sparsercnn_loss.py │ │ ├── ssd_loss.py │ │ ├── utils.py │ │ ├── varifocal_loss.py │ │ ├── yolo_loss.py │ │ ├── yolov5_loss.py │ │ └── yolov7_loss.py │ ├── mot │ │ ├── __init__.py │ │ ├── matching │ │ │ ├── __init__.py │ │ │ ├── deepsort_matching.py │ │ │ ├── jde_matching.py │ │ │ └── ocsort_matching.py │ │ ├── motion │ │ │ ├── __init__.py │ │ │ └── kalman_filter.py │ │ ├── tracker │ │ │ ├── __init__.py │ │ │ ├── base_jde_tracker.py │ │ │ ├── base_sde_tracker.py │ │ │ ├── deepsort_tracker.py │ │ │ ├── jde_tracker.py │ │ │ └── ocsort_tracker.py │ │ ├── utils.py │ │ └── visualization.py │ ├── necks │ │ ├── __init__.py │ │ ├── bifpn.py │ │ ├── blazeface_fpn.py │ │ ├── centernet_fpn.py │ │ ├── csp_pan.py │ │ ├── custom_pan.py │ │ ├── es_pan.py │ │ ├── fpn.py │ │ ├── hrfpn.py │ │ ├── lc_pan.py │ │ ├── rtmdet_pafpn.py │ │ ├── ttf_fpn.py │ │ ├── yolo_fpn.py │ │ ├── yolov6_pafpn.py │ │ ├── yolov7_pafpn.py │ │ └── yolov8_pafpn.py │ ├── ops.py │ ├── post_process.py │ ├── proposal_generator │ │ ├── __init__.py │ │ ├── anchor_generator.py │ │ ├── proposal_generator.py │ │ ├── rpn_head.py │ │ ├── target.py │ │ └── target_layer.py │ ├── rbox_utils.py │ ├── reid │ │ ├── __init__.py │ │ ├── fairmot_embedding_head.py │ │ ├── jde_embedding_head.py │ │ ├── pplcnet_embedding.py │ │ ├── pyramidal_embedding.py │ │ ├── resnet.py │ │ └── resnet_embedding.py │ ├── shape_spec.py │ ├── tests │ │ ├── __init__.py │ │ ├── imgs │ │ │ ├── coco2017_val2017_000000000139.jpg │ │ │ └── coco2017_val2017_000000000724.jpg │ │ ├── test_architectures.py │ │ ├── test_base.py │ │ ├── test_mstest.py │ │ ├── test_ops.py │ │ └── test_yolov3_loss.py │ └── transformers │ │ ├── __init__.py │ │ ├── deformable_transformer.py │ │ ├── detr_transformer.py │ │ ├── dino_transformer.py │ │ ├── group_detr_transformer.py │ │ ├── hybrid_encoder.py │ │ ├── mask_dino_transformer.py │ │ ├── matchers.py │ │ ├── petr_transformer.py │ │ ├── position_encoding.py │ │ ├── rtdetr_transformer.py │ │ └── utils.py ├── optimizer │ ├── __init__.py │ ├── adamw.py │ ├── ema.py │ └── optimizer.py ├── process │ ├── __init__.py │ ├── infer │ │ ├── __init__.py │ │ └── detection_predict.py │ └── transform │ │ ├── __init__.py │ │ ├── convert_paddle_config.py │ │ ├── convert_paddle_detection_to_torch.py │ │ └── convert_paddle_to_torch_base.py └── utils │ ├── __init__.py │ ├── base_utils.py │ ├── colormap.py │ ├── common_utils.py │ ├── constant.py │ ├── detection_checkpoint_utils.py │ ├── detection_cli_utils.py │ ├── detection_common_utils.py │ ├── detection_download.py │ ├── detection_infer_utils.py │ ├── detection_visualizer.py │ ├── env.py │ ├── file_utils.py │ ├── logger_utils.py │ ├── match_utils.py │ ├── model_utils.py │ ├── time_utils.py │ └── voc_utils.py ├── requirements.txt ├── setup.py └── tests └── process ├── __init__.py └── infer ├── __init__.py └── run_detection_infer.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea 2 | 3 | -------------------------------------------------------------------------------- /configs/cascade_rcnn/_base_/optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [8, 11] 9 | - !LinearWarmup 10 | start_factor: 0.001 11 | steps: 1000 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0001 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_instance.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1x.yml', 5 | '_base_/cascade_mask_rcnn_r50_fpn.yml', 6 | '_base_/cascade_mask_fpn_reader.yml', 7 | ] 8 | weights: output/cascade_mask_rcnn_r50_fpn_1x_coco/model_final 9 | -------------------------------------------------------------------------------- /configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_instance.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1x.yml', 5 | '_base_/cascade_mask_rcnn_r50_fpn.yml', 6 | '_base_/cascade_mask_fpn_reader.yml', 7 | ] 8 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_ssld_v2_pretrained.pdparams 9 | weights: output/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco/model_final 10 | 11 | ResNet: 12 | depth: 50 13 | variant: d 14 | norm_type: bn 15 | freeze_at: 0 16 | return_idx: [0,1,2,3] 17 | num_stages: 4 18 | lr_mult_list: [0.05, 0.05, 0.1, 0.15] 19 | -------------------------------------------------------------------------------- /configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1x.yml', 5 | '_base_/cascade_rcnn_r50_fpn.yml', 6 | '_base_/cascade_fpn_reader.yml', 7 | ] 8 | weights: output/cascade_rcnn_r50_fpn_1x_coco/model_final 9 | -------------------------------------------------------------------------------- /configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1x.yml', 5 | '_base_/cascade_rcnn_r50_fpn.yml', 6 | '_base_/cascade_fpn_reader.yml', 7 | ] 8 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_ssld_v2_pretrained.pdparams 9 | weights: output/cascade_rcnn_r50_vd_fpn_ssld_1x_coco/model_final 10 | 11 | ResNet: 12 | depth: 50 13 | variant: d 14 | norm_type: bn 15 | freeze_at: 0 16 | return_idx: [0,1,2,3] 17 | num_stages: 4 18 | lr_mult_list: [0.05, 0.05, 0.1, 0.15] 19 | -------------------------------------------------------------------------------- /configs/centernet/_base_/centernet_dla34.yml: -------------------------------------------------------------------------------- 1 | architecture: CenterNet 2 | pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/pretrained/DLA34_pretrain.pdparams 3 | 4 | CenterNet: 5 | backbone: DLA 6 | neck: CenterNetDLAFPN 7 | head: CenterNetHead 8 | post_process: CenterNetPostProcess 9 | 10 | DLA: 11 | depth: 34 12 | 13 | CenterNetDLAFPN: 14 | down_ratio: 4 15 | 16 | CenterNetHead: 17 | head_planes: 256 18 | regress_ltrb: False 19 | 20 | CenterNetPostProcess: 21 | max_per_img: 100 22 | regress_ltrb: False 23 | -------------------------------------------------------------------------------- /configs/centernet/_base_/optimizer_140e.yml: -------------------------------------------------------------------------------- 1 | epoch: 140 2 | 3 | LearningRate: 4 | base_lr: 0.0005 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [90, 120] 9 | use_warmup: False 10 | 11 | OptimizerBuilder: 12 | optimizer: 13 | type: Adam 14 | regularizer: NULL 15 | -------------------------------------------------------------------------------- /configs/centernet/centernet_dla34_140e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_140e.yml', 5 | '_base_/centernet_dla34.yml', 6 | '_base_/centernet_reader.yml', 7 | ] 8 | 9 | weights: output/centernet_dla34_140e_coco/model_final 10 | -------------------------------------------------------------------------------- /configs/centernet/centernet_mbv1_140e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'centernet_r50_140e_coco.yml' 3 | ] 4 | 5 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/MobileNetV1_pretrained.pdparams 6 | weights: output/centernet_mbv1_140e_coco/model_final 7 | 8 | CenterNet: 9 | backbone: MobileNet 10 | neck: CenterNetDLAFPN 11 | head: CenterNetHead 12 | post_process: CenterNetPostProcess 13 | 14 | MobileNet: 15 | scale: 1. 16 | with_extra_blocks: false 17 | extra_block_filters: [] 18 | feature_maps: [3, 5, 11, 13] 19 | 20 | TrainReader: 21 | batch_size: 32 22 | -------------------------------------------------------------------------------- /configs/centernet/centernet_mbv3_large_140e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'centernet_r50_140e_coco.yml' 3 | ] 4 | 5 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/MobileNetV3_large_x1_0_ssld_pretrained.pdparams 6 | weights: output/centernet_mbv3_large_140e_coco/model_final 7 | 8 | CenterNet: 9 | backbone: MobileNetV3 10 | neck: CenterNetDLAFPN 11 | head: CenterNetHead 12 | post_process: CenterNetPostProcess 13 | 14 | MobileNetV3: 15 | model_name: large 16 | scale: 1. 17 | with_extra_blocks: false 18 | extra_block_filters: [] 19 | feature_maps: [4, 7, 13, 16] 20 | 21 | TrainReader: 22 | batch_size: 32 23 | -------------------------------------------------------------------------------- /configs/centernet/centernet_r50_140e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_140e.yml', 5 | '_base_/centernet_r50.yml', 6 | '_base_/centernet_reader.yml', 7 | ] 8 | 9 | weights: output/centernet_r50_140e_coco/model_final 10 | -------------------------------------------------------------------------------- /configs/datasets/coco_detection.yml: -------------------------------------------------------------------------------- 1 | metric: COCO 2 | num_classes: 80 3 | 4 | TrainDataset: 5 | name: COCODataSet 6 | image_dir: train2017 7 | anno_path: annotations/instances_train2017.json 8 | dataset_dir: dataset/coco 9 | data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd'] 10 | 11 | EvalDataset: 12 | name: COCODataSet 13 | image_dir: val2017 14 | anno_path: annotations/instances_val2017.json 15 | dataset_dir: dataset/coco 16 | allow_empty: true 17 | 18 | TestDataset: 19 | name: ImageFolder 20 | anno_path: annotations/instances_val2017.json # also support txt (like VOC's label_list.txt) 21 | dataset_dir: dataset/coco # if set, anno_path will be 'dataset_dir/anno_path' 22 | -------------------------------------------------------------------------------- /configs/datasets/coco_instance.yml: -------------------------------------------------------------------------------- 1 | metric: COCO 2 | num_classes: 80 3 | 4 | TrainDataset: 5 | name: COCODataSet 6 | image_dir: train2017 7 | anno_path: annotations/instances_train2017.json 8 | dataset_dir: dataset/coco 9 | data_fields: ['image', 'gt_bbox', 'gt_class', 'gt_poly', 'is_crowd'] 10 | 11 | EvalDataset: 12 | name: COCODataSet 13 | image_dir: val2017 14 | anno_path: annotations/instances_val2017.json 15 | dataset_dir: dataset/coco 16 | 17 | TestDataset: 18 | name: ImageFolder 19 | anno_path: annotations/instances_val2017.json # also support txt (like VOC's label_list.txt) 20 | dataset_dir: dataset/coco # if set, anno_path will be 'dataset_dir/anno_path' 21 | -------------------------------------------------------------------------------- /configs/datasets/dota.yml: -------------------------------------------------------------------------------- 1 | metric: RBOX 2 | num_classes: 15 3 | 4 | TrainDataset: 5 | !COCODataSet 6 | image_dir: trainval1024/images 7 | anno_path: trainval1024/DOTA_trainval1024.json 8 | dataset_dir: dataset/dota/ 9 | data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd', 'gt_poly'] 10 | 11 | EvalDataset: 12 | !COCODataSet 13 | image_dir: trainval1024/images 14 | anno_path: trainval1024/DOTA_trainval1024.json 15 | dataset_dir: dataset/dota/ 16 | data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd', 'gt_poly'] 17 | 18 | TestDataset: 19 | !ImageFolder 20 | anno_path: test1024/DOTA_test1024.json 21 | dataset_dir: dataset/dota/ 22 | -------------------------------------------------------------------------------- /configs/datasets/dota_ms.yml: -------------------------------------------------------------------------------- 1 | metric: RBOX 2 | num_classes: 15 3 | 4 | TrainDataset: 5 | !COCODataSet 6 | image_dir: trainval1024/images 7 | anno_path: trainval1024/DOTA_trainval1024.json 8 | dataset_dir: dataset/dota_ms/ 9 | data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd', 'gt_poly'] 10 | 11 | EvalDataset: 12 | !COCODataSet 13 | image_dir: trainval1024/images 14 | anno_path: trainval1024/DOTA_trainval1024.json 15 | dataset_dir: dataset/dota_ms/ 16 | data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd', 'gt_poly'] 17 | 18 | TestDataset: 19 | !ImageFolder 20 | anno_path: test1024/DOTA_test1024.json 21 | dataset_dir: dataset/dota_ms/ 22 | -------------------------------------------------------------------------------- /configs/datasets/objects365_detection.yml: -------------------------------------------------------------------------------- 1 | metric: COCO 2 | num_classes: 365 3 | 4 | TrainDataset: 5 | !COCODataSet 6 | image_dir: train 7 | anno_path: annotations/zhiyuan_objv2_train.json 8 | dataset_dir: dataset/objects365 9 | data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd'] 10 | 11 | EvalDataset: 12 | !COCODataSet 13 | image_dir: val 14 | anno_path: annotations/zhiyuan_objv2_val.json 15 | dataset_dir: dataset/objects365 16 | allow_empty: true 17 | 18 | TestDataset: 19 | !ImageFolder 20 | anno_path: annotations/zhiyuan_objv2_val.json 21 | dataset_dir: dataset/objects365/ 22 | -------------------------------------------------------------------------------- /configs/datasets/roadsign_voc.yml: -------------------------------------------------------------------------------- 1 | metric: VOC 2 | map_type: integral 3 | num_classes: 4 4 | 5 | TrainDataset: 6 | name: VOCDataSet 7 | dataset_dir: dataset/roadsign_voc 8 | anno_path: train.txt 9 | label_list: label_list.txt 10 | data_fields: ['image', 'gt_bbox', 'gt_class', 'difficult'] 11 | 12 | EvalDataset: 13 | name: VOCDataSet 14 | dataset_dir: dataset/roadsign_voc 15 | anno_path: valid.txt 16 | label_list: label_list.txt 17 | data_fields: ['image', 'gt_bbox', 'gt_class', 'difficult'] 18 | 19 | TestDataset: 20 | name: ImageFolder 21 | anno_path: dataset/roadsign_voc/label_list.txt 22 | -------------------------------------------------------------------------------- /configs/datasets/spine_coco.yml: -------------------------------------------------------------------------------- 1 | metric: RBOX 2 | num_classes: 9 3 | 4 | TrainDataset: 5 | !COCODataSet 6 | image_dir: images 7 | anno_path: annotations/train.json 8 | dataset_dir: dataset/spine_coco 9 | data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd', 'gt_poly'] 10 | 11 | EvalDataset: 12 | !COCODataSet 13 | image_dir: images 14 | anno_path: annotations/valid.json 15 | dataset_dir: dataset/spine_coco 16 | data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd', 'gt_poly'] 17 | 18 | TestDataset: 19 | !ImageFolder 20 | anno_path: annotations/valid.json 21 | dataset_dir: dataset/spine_coco 22 | -------------------------------------------------------------------------------- /configs/datasets/visdrone_detection.yml: -------------------------------------------------------------------------------- 1 | metric: COCO 2 | num_classes: 10 3 | 4 | TrainDataset: 5 | !COCODataSet 6 | image_dir: VisDrone2019-DET-train 7 | anno_path: train.json 8 | dataset_dir: dataset/visdrone 9 | data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd'] 10 | 11 | EvalDataset: 12 | !COCODataSet 13 | image_dir: VisDrone2019-DET-val 14 | anno_path: val.json 15 | # image_dir: test_dev 16 | # anno_path: test_dev.json 17 | dataset_dir: dataset/visdrone 18 | 19 | TestDataset: 20 | !ImageFolder 21 | anno_path: val.json 22 | dataset_dir: dataset/visdrone 23 | -------------------------------------------------------------------------------- /configs/datasets/voc.yml: -------------------------------------------------------------------------------- 1 | metric: VOC 2 | map_type: 11point 3 | num_classes: 20 4 | 5 | TrainDataset: 6 | name: VOCDataSet 7 | dataset_dir: dataset/voc 8 | anno_path: trainval.txt 9 | label_list: label_list.txt 10 | data_fields: ['image', 'gt_bbox', 'gt_class', 'difficult'] 11 | 12 | EvalDataset: 13 | name: VOCDataSet 14 | dataset_dir: dataset/voc 15 | anno_path: test.txt 16 | label_list: label_list.txt 17 | data_fields: ['image', 'gt_bbox', 'gt_class', 'difficult'] 18 | 19 | TestDataset: 20 | name: ImageFolder 21 | anno_path: dataset/voc/label_list.txt 22 | -------------------------------------------------------------------------------- /configs/datasets/wider_face.yml: -------------------------------------------------------------------------------- 1 | metric: WiderFace 2 | num_classes: 1 3 | 4 | TrainDataset: 5 | !WIDERFaceDataSet 6 | dataset_dir: dataset/wider_face 7 | anno_path: wider_face_split/wider_face_train_bbx_gt.txt 8 | image_dir: WIDER_train/images 9 | data_fields: ['image', 'gt_bbox', 'gt_class'] 10 | 11 | EvalDataset: 12 | !WIDERFaceDataSet 13 | dataset_dir: dataset/wider_face 14 | anno_path: wider_face_split/wider_face_val_bbx_gt.txt 15 | image_dir: WIDER_val/images 16 | data_fields: ['image'] 17 | 18 | TestDataset: 19 | !ImageFolder 20 | use_default_label: true 21 | -------------------------------------------------------------------------------- /configs/dcn/cascade_rcnn_dcn_r50_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '../cascade_rcnn/_base_/optimizer_1x.yml', 5 | '../cascade_rcnn/_base_/cascade_rcnn_r50_fpn.yml', 6 | '../cascade_rcnn/_base_/cascade_fpn_reader.yml', 7 | ] 8 | weights: output/cascade_rcnn_dcn_r50_fpn_1x_coco/model_final 9 | 10 | ResNet: 11 | depth: 50 12 | norm_type: bn 13 | freeze_at: 0 14 | return_idx: [0,1,2,3] 15 | num_stages: 4 16 | dcn_v2_stages: [1,2,3] 17 | -------------------------------------------------------------------------------- /configs/dcn/cascade_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'cascade_rcnn_dcn_r50_fpn_1x_coco.yml', 3 | ] 4 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNeXt101_vd_64x4d_pretrained.pdparams 5 | weights: output/cascade_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco/model_final 6 | 7 | ResNet: 8 | depth: 101 9 | groups: 64 10 | base_width: 4 11 | variant: d 12 | norm_type: bn 13 | freeze_at: 0 14 | return_idx: [0,1,2,3] 15 | num_stages: 4 16 | dcn_v2_stages: [1,2,3] 17 | -------------------------------------------------------------------------------- /configs/dcn/faster_rcnn_dcn_r101_vd_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_dcn_r50_fpn_1x_coco.yml', 3 | ] 4 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet101_vd_pretrained.pdparams 5 | weights: output/faster_rcnn_dcn_r101_vd_fpn_1x_coco/model_final 6 | 7 | ResNet: 8 | # index 0 stands for res2 9 | depth: 101 10 | variant: d 11 | norm_type: bn 12 | freeze_at: 0 13 | return_idx: [0,1,2,3] 14 | num_stages: 4 15 | dcn_v2_stages: [1,2,3] 16 | -------------------------------------------------------------------------------- /configs/dcn/faster_rcnn_dcn_r50_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '../faster_rcnn/_base_/optimizer_1x.yml', 5 | '../faster_rcnn/_base_/faster_rcnn_r50_fpn.yml', 6 | '../faster_rcnn/_base_/faster_fpn_reader.yml', 7 | ] 8 | weights: output/faster_rcnn_dcn_r50_fpn_1x_coco/model_final 9 | 10 | ResNet: 11 | depth: 50 12 | norm_type: bn 13 | freeze_at: 0 14 | return_idx: [0,1,2,3] 15 | num_stages: 4 16 | dcn_v2_stages: [1,2,3] 17 | -------------------------------------------------------------------------------- /configs/dcn/faster_rcnn_dcn_r50_vd_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_dcn_r50_fpn_1x_coco.yml', 3 | ] 4 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_pretrained.pdparams 5 | weights: output/faster_rcnn_dcn_r50_vd_fpn_2x_coco/model_final 6 | 7 | ResNet: 8 | # index 0 stands for res2 9 | depth: 50 10 | variant: d 11 | norm_type: bn 12 | freeze_at: 0 13 | return_idx: [0,1,2,3] 14 | num_stages: 4 15 | dcn_v2_stages: [1,2,3] 16 | -------------------------------------------------------------------------------- /configs/dcn/faster_rcnn_dcn_r50_vd_fpn_2x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_dcn_r50_fpn_1x_coco.yml', 3 | ] 4 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_pretrained.pdparams 5 | weights: output/faster_rcnn_dcn_r50_vd_fpn_2x_coco/model_final 6 | 7 | ResNet: 8 | # index 0 stands for res2 9 | depth: 50 10 | variant: d 11 | norm_type: bn 12 | freeze_at: 0 13 | return_idx: [0,1,2,3] 14 | num_stages: 4 15 | dcn_v2_stages: [1,2,3] 16 | 17 | epoch: 24 18 | LearningRate: 19 | base_lr: 0.01 20 | schedulers: 21 | - !PiecewiseDecay 22 | gamma: 0.1 23 | milestones: [16, 22] 24 | - !LinearWarmup 25 | start_factor: 0.1 26 | steps: 1000 27 | -------------------------------------------------------------------------------- /configs/dcn/faster_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_dcn_r50_fpn_1x_coco.yml', 3 | ] 4 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNeXt101_vd_64x4d_pretrained.pdparams 5 | weights: output/faster_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco/model_final 6 | 7 | ResNet: 8 | # for ResNeXt: groups, base_width, base_channels 9 | depth: 101 10 | groups: 64 11 | base_width: 4 12 | variant: d 13 | norm_type: bn 14 | freeze_at: 0 15 | return_idx: [0,1,2,3] 16 | num_stages: 4 17 | dcn_v2_stages: [1,2,3] 18 | -------------------------------------------------------------------------------- /configs/dcn/mask_rcnn_dcn_r101_vd_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'mask_rcnn_dcn_r50_fpn_1x_coco.yml', 3 | ] 4 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet101_vd_pretrained.pdparams 5 | weights: output/mask_rcnn_dcn_r101_vd_fpn_1x_coco/model_final 6 | 7 | ResNet: 8 | # index 0 stands for res2 9 | depth: 101 10 | variant: d 11 | norm_type: bn 12 | freeze_at: 0 13 | return_idx: [0,1,2,3] 14 | num_stages: 4 15 | dcn_v2_stages: [1,2,3] 16 | -------------------------------------------------------------------------------- /configs/dcn/mask_rcnn_dcn_r50_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_instance.yml', 3 | '../runtime.yml', 4 | '../mask_rcnn/_base_/optimizer_1x.yml', 5 | '../mask_rcnn/_base_/mask_rcnn_r50_fpn.yml', 6 | '../mask_rcnn/_base_/mask_fpn_reader.yml', 7 | ] 8 | weights: output/mask_rcnn_dcn_r50_fpn_1x_coco/model_final 9 | 10 | ResNet: 11 | depth: 50 12 | norm_type: bn 13 | freeze_at: 0 14 | return_idx: [0,1,2,3] 15 | num_stages: 4 16 | dcn_v2_stages: [1,2,3] 17 | -------------------------------------------------------------------------------- /configs/dcn/mask_rcnn_dcn_r50_vd_fpn_2x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'mask_rcnn_dcn_r50_fpn_1x_coco.yml', 3 | ] 4 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_pretrained.pdparams 5 | weights: output/mask_rcnn_dcn_r50_vd_fpn_2x_coco/model_final 6 | 7 | ResNet: 8 | # index 0 stands for res2 9 | depth: 50 10 | variant: d 11 | norm_type: bn 12 | freeze_at: 0 13 | return_idx: [0,1,2,3] 14 | num_stages: 4 15 | dcn_v2_stages: [1,2,3] 16 | 17 | epoch: 24 18 | LearningRate: 19 | base_lr: 0.01 20 | schedulers: 21 | - !PiecewiseDecay 22 | gamma: 0.1 23 | milestones: [16, 22] 24 | - !LinearWarmup 25 | start_factor: 0.1 26 | steps: 1000 27 | -------------------------------------------------------------------------------- /configs/dcn/mask_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'mask_rcnn_dcn_r50_fpn_1x_coco.yml', 3 | ] 4 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNeXt101_vd_64x4d_pretrained.pdparams 5 | weights: output/mask_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco/model_final 6 | 7 | ResNet: 8 | # for ResNeXt: groups, base_width, base_channels 9 | depth: 101 10 | variant: d 11 | groups: 64 12 | base_width: 4 13 | norm_type: bn 14 | freeze_at: 0 15 | return_idx: [0,1,2,3] 16 | num_stages: 4 17 | dcn_v2_stages: [1,2,3] 18 | -------------------------------------------------------------------------------- /configs/deformable_detr/_base_/deformable_optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 50 2 | 3 | LearningRate: 4 | base_lr: 0.0002 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [40] 9 | use_warmup: false 10 | 11 | OptimizerBuilder: 12 | clip_grad_by_norm: 0.1 13 | regularizer: false 14 | optimizer: 15 | type: AdamW 16 | weight_decay: 0.0001 17 | -------------------------------------------------------------------------------- /configs/deformable_detr/deformable_detr_r50_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/deformable_optimizer_1x.yml', 5 | '_base_/deformable_detr_r50.yml', 6 | '_base_/deformable_detr_reader.yml', 7 | ] 8 | weights: output/deformable_detr_r50_1x_coco/model_final 9 | find_unused_parameters: True 10 | -------------------------------------------------------------------------------- /configs/detr/_base_/optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 500 2 | 3 | LearningRate: 4 | base_lr: 0.0001 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [400] 9 | use_warmup: false 10 | 11 | OptimizerBuilder: 12 | clip_grad_by_norm: 0.1 13 | regularizer: false 14 | optimizer: 15 | type: AdamW 16 | weight_decay: 0.0001 17 | -------------------------------------------------------------------------------- /configs/detr/detr_r50_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1x.yml', 5 | '_base_/detr_r50.yml', 6 | '_base_/detr_reader.yml', 7 | ] 8 | weights: output/detr_r50_1x_coco/model_final 9 | find_unused_parameters: True 10 | -------------------------------------------------------------------------------- /configs/dino/_base_/optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.0001 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [11] 9 | use_warmup: false 10 | 11 | OptimizerBuilder: 12 | clip_grad_by_norm: 0.1 13 | regularizer: false 14 | optimizer: 15 | type: AdamW 16 | weight_decay: 0.0001 17 | -------------------------------------------------------------------------------- /configs/dino/_base_/optimizer_2x.yml: -------------------------------------------------------------------------------- 1 | epoch: 24 2 | 3 | LearningRate: 4 | base_lr: 0.0001 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [20] 9 | use_warmup: false 10 | 11 | OptimizerBuilder: 12 | clip_grad_by_norm: 0.1 13 | regularizer: false 14 | optimizer: 15 | type: AdamW 16 | weight_decay: 0.0001 17 | -------------------------------------------------------------------------------- /configs/dino/_base_/optimizer_3x.yml: -------------------------------------------------------------------------------- 1 | epoch: 36 2 | 3 | LearningRate: 4 | base_lr: 0.0001 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [33] 9 | use_warmup: false 10 | 11 | OptimizerBuilder: 12 | clip_grad_by_norm: 0.1 13 | regularizer: false 14 | optimizer: 15 | type: AdamW 16 | weight_decay: 0.0001 17 | -------------------------------------------------------------------------------- /configs/dino/dino_r50_4scale_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1x.yml', 5 | '_base_/dino_r50.yml', 6 | '_base_/dino_reader.yml', 7 | ] 8 | 9 | weights: output/dino_r50_4scale_1x_coco/model_final 10 | find_unused_parameters: True 11 | log_iter: 100 12 | -------------------------------------------------------------------------------- /configs/dino/dino_r50_4scale_2x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_2x.yml', 5 | '_base_/dino_r50.yml', 6 | '_base_/dino_reader.yml', 7 | ] 8 | 9 | weights: output/dino_r50_4scale_2x_coco/model_final 10 | find_unused_parameters: True 11 | log_iter: 100 12 | -------------------------------------------------------------------------------- /configs/face_detection/_base_/optimizer_1000e.yml: -------------------------------------------------------------------------------- 1 | epoch: 1000 2 | 3 | LearningRate: 4 | base_lr: 0.001 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: 9 | - 333 10 | - 800 11 | - !LinearWarmup 12 | start_factor: 0.3333333333333333 13 | steps: 500 14 | 15 | OptimizerBuilder: 16 | optimizer: 17 | momentum: 0.0 18 | type: RMSProp 19 | regularizer: 20 | factor: 0.0005 21 | type: L2 22 | -------------------------------------------------------------------------------- /configs/face_detection/blazeface_1000e.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/wider_face.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1000e.yml', 5 | '_base_/blazeface.yml', 6 | '_base_/face_reader.yml', 7 | ] 8 | weights: output/blazeface_1000e/model_final 9 | multi_scale_eval: True 10 | -------------------------------------------------------------------------------- /configs/face_detection/blazeface_fpn_ssh_1000e.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/wider_face.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1000e.yml', 5 | '_base_/blazeface_fpn.yml', 6 | '_base_/face_reader.yml', 7 | ] 8 | weights: output/blazeface_fpn_ssh_1000e/model_final 9 | multi_scale_eval: True 10 | -------------------------------------------------------------------------------- /configs/faster_rcnn/_base_/optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [8, 11] 9 | - !LinearWarmup 10 | start_factor: 0.1 11 | steps: 1000 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0001 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/faster_rcnn/_base_/optimizer_swin_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.0001 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [8, 11] 9 | - !LinearWarmup 10 | start_factor: 0.1 11 | steps: 1000 12 | 13 | OptimizerBuilder: 14 | clip_grad_by_norm: 1.0 15 | optimizer: 16 | type: AdamW 17 | weight_decay: 0.05 18 | param_groups: 19 | - params: ['absolute_pos_embed', 'relative_position_bias_table', 'norm'] 20 | weight_decay: 0.0 21 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_r101_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_r50_1x_coco.yml', 3 | ] 4 | 5 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet101_pretrained.pdparams 6 | weights: output/faster_rcnn_r101_1x_coco/model_final 7 | 8 | ResNet: 9 | # index 0 stands for res2 10 | depth: 101 11 | norm_type: bn 12 | freeze_at: 0 13 | return_idx: [2] 14 | num_stages: 3 15 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_r50_fpn_1x_coco.yml', 3 | ] 4 | 5 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet101_pretrained.pdparams 6 | weights: output/faster_rcnn_r101_fpn_1x_coco/model_final 7 | 8 | ResNet: 9 | # index 0 stands for res2 10 | depth: 101 11 | norm_type: bn 12 | freeze_at: 0 13 | return_idx: [0,1,2,3] 14 | num_stages: 4 15 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_r101_fpn_2x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_r50_fpn_1x_coco.yml', 3 | ] 4 | 5 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet101_pretrained.pdparams 6 | weights: output/faster_rcnn_r101_fpn_2x_coco/model_final 7 | 8 | ResNet: 9 | # index 0 stands for res2 10 | depth: 101 11 | norm_type: bn 12 | freeze_at: 0 13 | return_idx: [0,1,2,3] 14 | num_stages: 4 15 | 16 | epoch: 24 17 | LearningRate: 18 | base_lr: 0.01 19 | schedulers: 20 | - !PiecewiseDecay 21 | gamma: 0.1 22 | milestones: [16, 22] 23 | - !LinearWarmup 24 | start_factor: 0.1 25 | steps: 1000 26 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_r50_fpn_1x_coco.yml', 3 | ] 4 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet101_vd_pretrained.pdparams 5 | weights: output/faster_rcnn_r101_vd_fpn_1x_coco/model_final 6 | 7 | ResNet: 8 | # index 0 stands for res2 9 | depth: 101 10 | variant: d 11 | norm_type: bn 12 | freeze_at: 0 13 | return_idx: [0,1,2,3] 14 | num_stages: 4 15 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_r101_vd_fpn_2x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_r50_fpn_1x_coco.yml', 3 | ] 4 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet101_vd_pretrained.pdparams 5 | weights: output/faster_rcnn_r101_vd_fpn_2x_coco/model_final 6 | 7 | ResNet: 8 | # index 0 stands for res2 9 | depth: 101 10 | variant: d 11 | norm_type: bn 12 | freeze_at: 0 13 | return_idx: [0,1,2,3] 14 | num_stages: 4 15 | 16 | epoch: 24 17 | LearningRate: 18 | base_lr: 0.01 19 | schedulers: 20 | - !PiecewiseDecay 21 | gamma: 0.1 22 | milestones: [16, 22] 23 | - !LinearWarmup 24 | start_factor: 0.1 25 | steps: 1000 26 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_r50_fpn_1x_coco.yml', 3 | ] 4 | 5 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet34_pretrained.pdparams 6 | weights: output/faster_rcnn_r34_fpn_1x_coco/model_final 7 | 8 | ResNet: 9 | # index 0 stands for res2 10 | depth: 34 11 | norm_type: bn 12 | freeze_at: 0 13 | return_idx: [0,1,2,3] 14 | num_stages: 4 15 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_r50_fpn_1x_coco.yml', 3 | ] 4 | 5 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet34_vd_pretrained.pdparams 6 | weights: output/faster_rcnn_r34_vd_fpn_1x_coco/model_final 7 | 8 | ResNet: 9 | # index 0 stands for res2 10 | depth: 34 11 | variant: d 12 | norm_type: bn 13 | freeze_at: 0 14 | return_idx: [0,1,2,3] 15 | num_stages: 4 16 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_r50_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1x.yml', 5 | '_base_/faster_rcnn_r50.yml', 6 | '_base_/faster_reader.yml', 7 | ] 8 | weights: output/faster_rcnn_r50_1x_coco/model_final 9 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1x.yml', 5 | '_base_/faster_rcnn_r50_fpn.yml', 6 | '_base_/faster_fpn_reader.yml', 7 | ] 8 | weights: output/faster_rcnn_r50_fpn_1x_coco/model_final 9 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_r50_fpn_2x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_r50_fpn_1x_coco.yml', 3 | ] 4 | weights: output/faster_rcnn_r50_fpn_2x_coco/model_final 5 | 6 | epoch: 24 7 | LearningRate: 8 | base_lr: 0.01 9 | schedulers: 10 | - !PiecewiseDecay 11 | gamma: 0.1 12 | milestones: [16, 22] 13 | - !LinearWarmup 14 | start_factor: 0.1 15 | steps: 1000 16 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_r50_1x_coco.yml', 3 | ] 4 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_pretrained.pdparams 5 | weights: output/faster_rcnn_r50_vd_1x_coco/model_final 6 | 7 | ResNet: 8 | # index 0 stands for res2 9 | depth: 50 10 | variant: d 11 | norm_type: bn 12 | freeze_at: 0 13 | return_idx: [2] 14 | num_stages: 3 15 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_r50_fpn_1x_coco.yml', 3 | ] 4 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_pretrained.pdparams 5 | weights: output/faster_rcnn_r50_vd_fpn_1x_coco/model_final 6 | 7 | ResNet: 8 | # index 0 stands for res2 9 | depth: 50 10 | variant: d 11 | norm_type: bn 12 | freeze_at: 0 13 | return_idx: [0,1,2,3] 14 | num_stages: 4 15 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_r50_vd_fpn_2x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_r50_fpn_1x_coco.yml', 3 | ] 4 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_pretrained.pdparams 5 | weights: output/faster_rcnn_r50_vd_fpn_2x_coco/model_final 6 | 7 | ResNet: 8 | # index 0 stands for res2 9 | depth: 50 10 | variant: d 11 | norm_type: bn 12 | freeze_at: 0 13 | return_idx: [0,1,2,3] 14 | num_stages: 4 15 | 16 | epoch: 24 17 | LearningRate: 18 | base_lr: 0.01 19 | schedulers: 20 | - !PiecewiseDecay 21 | gamma: 0.1 22 | milestones: [16, 22] 23 | - !LinearWarmup 24 | start_factor: 0.1 25 | steps: 1000 26 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_swin_1x.yml', 5 | '_base_/faster_rcnn_swin_tiny_fpn.yml', 6 | '_base_/faster_rcnn_swin_reader.yml', 7 | ] 8 | weights: output/faster_rcnn_swin_tiny_fpn_1x_coco/model_final 9 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_2x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_swin_tiny_fpn_1x_coco.yml', 3 | ] 4 | weights: output/faster_rcnn_swin_tiny_fpn_2x_coco/model_final 5 | 6 | epoch: 24 7 | 8 | LearningRate: 9 | base_lr: 0.0001 10 | schedulers: 11 | - !PiecewiseDecay 12 | gamma: 0.1 13 | milestones: [16, 22] 14 | - !LinearWarmup 15 | start_factor: 0.1 16 | steps: 1000 17 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_3x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_swin_tiny_fpn_1x_coco.yml', 3 | ] 4 | weights: output/faster_rcnn_swin_tiny_fpn_3x_coco/model_final 5 | 6 | epoch: 36 7 | 8 | LearningRate: 9 | base_lr: 0.0001 10 | schedulers: 11 | - !PiecewiseDecay 12 | gamma: 0.1 13 | milestones: [24, 33] 14 | - !LinearWarmup 15 | start_factor: 0.1 16 | steps: 1000 17 | 18 | OptimizerBuilder: 19 | clip_grad_by_norm: 1.0 20 | optimizer: 21 | type: AdamW 22 | weight_decay: 0.05 23 | -------------------------------------------------------------------------------- /configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'faster_rcnn_r50_fpn_1x_coco.yml', 3 | ] 4 | 5 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNeXt101_vd_64x4d_pretrained.pdparams 6 | weights: output/faster_rcnn_x101_vd_64x4d_fpn_1x_coco/model_final 7 | 8 | ResNet: 9 | # for ResNeXt: groups, base_width, base_channels 10 | depth: 101 11 | groups: 64 12 | base_width: 4 13 | variant: d 14 | norm_type: bn 15 | freeze_at: 0 16 | return_idx: [0,1,2,3] 17 | num_stages: 4 18 | -------------------------------------------------------------------------------- /configs/fcos/_base_/optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [8, 11] 9 | - !LinearWarmup 10 | start_factor: 0.3333333333333333 11 | steps: 500 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0001 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/fcos/fcos_dcn_r50_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/fcos_r50_fpn.yml', 5 | '_base_/optimizer_1x.yml', 6 | '_base_/fcos_reader.yml', 7 | ] 8 | 9 | weights: output/fcos_dcn_r50_fpn_1x_coco/model_final 10 | 11 | ResNet: 12 | dcn_v2_stages: [1, 2, 3] 13 | 14 | FCOSHead: 15 | fcos_feat: 16 | use_dcn: True 17 | -------------------------------------------------------------------------------- /configs/fcos/fcos_r50_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/fcos_r50_fpn.yml', 5 | '_base_/optimizer_1x.yml', 6 | '_base_/fcos_reader.yml', 7 | ] 8 | 9 | weights: output/fcos_r50_fpn_1x_coco/model_final 10 | -------------------------------------------------------------------------------- /configs/few-shot/_base_/optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [8, 11] 9 | - !LinearWarmup 10 | start_factor: 0.1 11 | steps: 1000 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0001 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/few-shot/_base_/optimizer_80e.yml: -------------------------------------------------------------------------------- 1 | epoch: 80 2 | 3 | LearningRate: 4 | base_lr: 0.001 5 | schedulers: 6 | - !CosineDecay 7 | max_epochs: 96 8 | - !LinearWarmup 9 | start_factor: 0. 10 | epochs: 5 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | momentum: 0.9 15 | type: Momentum 16 | regularizer: 17 | factor: 0.0005 18 | type: L2 19 | -------------------------------------------------------------------------------- /configs/gfl/_base_/optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [8, 11] 9 | - !LinearWarmup 10 | start_factor: 0.001 11 | steps: 500 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0001 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/gfl/gfl_r18vd_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/gfl_r50_fpn.yml', 5 | '_base_/optimizer_1x.yml', 6 | '_base_/gfl_reader.yml', 7 | ] 8 | 9 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet18_vd_pretrained.pdparams 10 | weights: output/gfl_r18vd_1x_coco/model_final 11 | find_unused_parameters: True 12 | 13 | ResNet: 14 | depth: 18 15 | variant: d 16 | norm_type: bn 17 | freeze_at: 0 18 | return_idx: [1,2,3] 19 | num_stages: 4 20 | -------------------------------------------------------------------------------- /configs/gfl/gfl_r34vd_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/gfl_r50_fpn.yml', 5 | '_base_/optimizer_1x.yml', 6 | '_base_/gfl_reader.yml', 7 | ] 8 | 9 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet34_vd_pretrained.pdparams 10 | weights: output/gfl_r34vd_1x_coco/model_final 11 | find_unused_parameters: True 12 | 13 | ResNet: 14 | depth: 34 15 | variant: d 16 | norm_type: bn 17 | freeze_at: 0 18 | return_idx: [1,2,3] 19 | num_stages: 4 20 | -------------------------------------------------------------------------------- /configs/gfl/gfl_r50_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/gfl_r50_fpn.yml', 5 | '_base_/optimizer_1x.yml', 6 | '_base_/gfl_reader.yml', 7 | ] 8 | 9 | weights: output/gfl_r50_fpn_1x_coco/model_final 10 | find_unused_parameters: True 11 | -------------------------------------------------------------------------------- /configs/gfl/gflv2_r50_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/gflv2_r50_fpn.yml', 5 | '_base_/optimizer_1x.yml', 6 | '_base_/gfl_reader.yml', 7 | ] 8 | 9 | weights: output/gflv2_r50_fpn_1x_coco/model_final 10 | find_unused_parameters: True 11 | -------------------------------------------------------------------------------- /configs/group_detr/_base_/optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.0001 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [11] 9 | use_warmup: false 10 | 11 | OptimizerBuilder: 12 | clip_grad_by_norm: 0.1 13 | regularizer: false 14 | optimizer: 15 | type: AdamW 16 | weight_decay: 0.0001 17 | -------------------------------------------------------------------------------- /configs/group_detr/group_dino_r50_4scale_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1x.yml', 5 | '_base_/group_dino_r50.yml', 6 | '_base_/dino_reader.yml', 7 | ] 8 | 9 | weights: output/group_dino_r50_4scale_1x_coco/model_final 10 | find_unused_parameters: True 11 | log_iter: 100 12 | -------------------------------------------------------------------------------- /configs/group_detr/group_dino_vit_huge_4scale_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1x.yml', 5 | '_base_/group_dino_vit_huge.yml', 6 | '_base_/dino_2000_reader.yml', 7 | ] 8 | 9 | weights: output/group_dino_vit_huge_4scale_1x_coco/model_final 10 | find_unused_parameters: True 11 | log_iter: 100 12 | -------------------------------------------------------------------------------- /configs/hrnet/faster_rcnn_hrnetv2p_w18_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | './_base_/faster_rcnn_hrnetv2p_w18.yml', 4 | '../faster_rcnn/_base_/optimizer_1x.yml', 5 | '../faster_rcnn/_base_/faster_fpn_reader.yml', 6 | '../runtime.yml', 7 | ] 8 | 9 | weights: output/faster_rcnn_hrnetv2p_w18_1x_coco/model_final 10 | epoch: 12 11 | 12 | LearningRate: 13 | base_lr: 0.02 14 | schedulers: 15 | - !PiecewiseDecay 16 | gamma: 0.1 17 | milestones: [8, 11] 18 | - !LinearWarmup 19 | start_factor: 0.1 20 | steps: 1000 21 | 22 | TrainReader: 23 | batch_size: 2 24 | -------------------------------------------------------------------------------- /configs/hrnet/faster_rcnn_hrnetv2p_w18_2x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | './_base_/faster_rcnn_hrnetv2p_w18.yml', 4 | '../faster_rcnn/_base_/optimizer_1x.yml', 5 | '../faster_rcnn/_base_/faster_fpn_reader.yml', 6 | '../runtime.yml', 7 | ] 8 | 9 | weights: output/faster_rcnn_hrnetv2p_w18_2x_coco/model_final 10 | epoch: 24 11 | 12 | LearningRate: 13 | base_lr: 0.02 14 | schedulers: 15 | - !PiecewiseDecay 16 | gamma: 0.1 17 | milestones: [16, 22] 18 | - !LinearWarmup 19 | start_factor: 0.1 20 | steps: 1000 21 | 22 | TrainReader: 23 | batch_size: 2 24 | -------------------------------------------------------------------------------- /configs/mask_rcnn/_base_/optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [8, 11] 9 | - !LinearWarmup 10 | start_factor: 0.001 11 | steps: 1000 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0001 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'mask_rcnn_r50_fpn_1x_coco.yml', 3 | ] 4 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet101_pretrained.pdparams 5 | weights: output/mask_rcnn_r101_fpn_1x_coco/model_final 6 | 7 | ResNet: 8 | # index 0 stands for res2 9 | depth: 101 10 | norm_type: bn 11 | freeze_at: 0 12 | return_idx: [0,1,2,3] 13 | num_stages: 4 14 | -------------------------------------------------------------------------------- /configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'mask_rcnn_r50_fpn_1x_coco.yml', 3 | ] 4 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet101_vd_pretrained.pdparams 5 | weights: output/mask_rcnn_r101_vd_fpn_1x_coco/model_final 6 | 7 | ResNet: 8 | # index 0 stands for res2 9 | depth: 101 10 | variant: d 11 | norm_type: bn 12 | freeze_at: 0 13 | return_idx: [0,1,2,3] 14 | num_stages: 4 15 | -------------------------------------------------------------------------------- /configs/mask_rcnn/mask_rcnn_r50_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_instance.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1x.yml', 5 | '_base_/mask_rcnn_r50.yml', 6 | '_base_/mask_reader.yml', 7 | ] 8 | weights: output/mask_rcnn_r50_1x_coco/model_final 9 | -------------------------------------------------------------------------------- /configs/mask_rcnn/mask_rcnn_r50_2x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'mask_rcnn_r50_1x_coco.yml', 3 | ] 4 | weights: output/mask_rcnn_r50_2x_coco/model_final 5 | 6 | epoch: 24 7 | LearningRate: 8 | base_lr: 0.01 9 | schedulers: 10 | - !PiecewiseDecay 11 | gamma: 0.1 12 | milestones: [16, 22] 13 | - !LinearWarmup 14 | start_factor: 0.3333333333333333 15 | steps: 500 16 | -------------------------------------------------------------------------------- /configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_instance.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1x.yml', 5 | '_base_/mask_rcnn_r50_fpn.yml', 6 | '_base_/mask_fpn_reader.yml', 7 | ] 8 | weights: output/mask_rcnn_r50_fpn_1x_coco/model_final 9 | -------------------------------------------------------------------------------- /configs/mask_rcnn/mask_rcnn_r50_fpn_2x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'mask_rcnn_r50_fpn_1x_coco.yml', 3 | ] 4 | weights: output/mask_rcnn_r50_fpn_2x_coco/model_final 5 | 6 | epoch: 24 7 | LearningRate: 8 | base_lr: 0.01 9 | schedulers: 10 | - !PiecewiseDecay 11 | gamma: 0.1 12 | milestones: [16, 22] 13 | - !LinearWarmup 14 | start_factor: 0.3333333333333333 15 | steps: 500 16 | -------------------------------------------------------------------------------- /configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'mask_rcnn_r50_fpn_1x_coco.yml', 3 | ] 4 | 5 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_pretrained.pdparams 6 | weights: output/mask_rcnn_r50_vd_fpn_1x_coco/model_final 7 | 8 | ResNet: 9 | # index 0 stands for res2 10 | depth: 50 11 | variant: d 12 | norm_type: bn 13 | freeze_at: 0 14 | return_idx: [0,1,2,3] 15 | num_stages: 4 16 | -------------------------------------------------------------------------------- /configs/mask_rcnn/mask_rcnn_r50_vd_fpn_2x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'mask_rcnn_r50_fpn_1x_coco.yml', 3 | ] 4 | 5 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_pretrained.pdparams 6 | weights: output/mask_rcnn_r50_vd_fpn_2x_coco/model_final 7 | 8 | ResNet: 9 | # index 0 stands for res2 10 | depth: 50 11 | variant: d 12 | norm_type: bn 13 | freeze_at: 0 14 | return_idx: [0,1,2,3] 15 | num_stages: 4 16 | 17 | epoch: 24 18 | LearningRate: 19 | base_lr: 0.01 20 | schedulers: 21 | - !PiecewiseDecay 22 | gamma: 0.1 23 | milestones: [16, 22] 24 | - !LinearWarmup 25 | start_factor: 0.3333333333333333 26 | steps: 500 27 | -------------------------------------------------------------------------------- /configs/mot/bytetrack/README.md: -------------------------------------------------------------------------------- 1 | README_cn.md -------------------------------------------------------------------------------- /configs/mot/bytetrack/detector/README.md: -------------------------------------------------------------------------------- 1 | README_cn.md -------------------------------------------------------------------------------- /configs/mot/centertrack/README.md: -------------------------------------------------------------------------------- 1 | README_cn.md -------------------------------------------------------------------------------- /configs/mot/centertrack/_base_/optimizer_70e.yml: -------------------------------------------------------------------------------- 1 | epoch: 70 2 | 3 | LearningRate: 4 | base_lr: 0.000125 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [60] 9 | use_warmup: False 10 | 11 | OptimizerBuilder: 12 | optimizer: 13 | type: Adam 14 | regularizer: NULL 15 | -------------------------------------------------------------------------------- /configs/mot/deepsort/README.md: -------------------------------------------------------------------------------- 1 | README_cn.md -------------------------------------------------------------------------------- /configs/mot/fairmot/_base_/optimizer_30e.yml: -------------------------------------------------------------------------------- 1 | epoch: 30 2 | 3 | LearningRate: 4 | base_lr: 0.0001 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [20,] 9 | use_warmup: False 10 | 11 | OptimizerBuilder: 12 | optimizer: 13 | type: Adam 14 | regularizer: NULL 15 | -------------------------------------------------------------------------------- /configs/mot/fairmot/_base_/optimizer_30e_momentum.yml: -------------------------------------------------------------------------------- 1 | epoch: 30 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [15, 22] 9 | use_warmup: True 10 | - !ExpWarmup 11 | steps: 1000 12 | power: 4 13 | 14 | OptimizerBuilder: 15 | optimizer: 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0001 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/mot/fairmot/fairmot_dla34_30e_1088x608.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/mot.yml', 3 | '../../runtime.yml', 4 | '_base_/optimizer_30e.yml', 5 | '_base_/fairmot_dla34.yml', 6 | '_base_/fairmot_reader_1088x608.yml', 7 | ] 8 | 9 | weights: output/fairmot_dla34_30e_1088x608/model_final 10 | -------------------------------------------------------------------------------- /configs/mot/fairmot/fairmot_dla34_30e_576x320.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/mot.yml', 3 | '../../runtime.yml', 4 | '_base_/optimizer_30e.yml', 5 | '_base_/fairmot_dla34.yml', 6 | '_base_/fairmot_reader_576x320.yml', 7 | ] 8 | 9 | weights: output/fairmot_dla34_30e_576x320/model_final 10 | -------------------------------------------------------------------------------- /configs/mot/fairmot/fairmot_dla34_30e_864x480.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/mot.yml', 3 | '../../runtime.yml', 4 | '_base_/optimizer_30e.yml', 5 | '_base_/fairmot_dla34.yml', 6 | '_base_/fairmot_reader_864x480.yml', 7 | ] 8 | 9 | weights: output/fairmot_dla34_30e_864x480/model_final 10 | -------------------------------------------------------------------------------- /configs/mot/headtracking21/README.md: -------------------------------------------------------------------------------- 1 | README_cn.md -------------------------------------------------------------------------------- /configs/mot/jde/_base_/optimizer_30e.yml: -------------------------------------------------------------------------------- 1 | epoch: 30 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [15, 22] 9 | use_warmup: True 10 | - !ExpWarmup 11 | steps: 1000 12 | power: 4 13 | 14 | OptimizerBuilder: 15 | optimizer: 16 | momentum: 0.9 17 | type: Momentum 18 | regularizer: 19 | factor: 0.0001 20 | type: L2 21 | -------------------------------------------------------------------------------- /configs/mot/jde/_base_/optimizer_60e.yml: -------------------------------------------------------------------------------- 1 | epoch: 60 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [30, 44] 9 | use_warmup: True 10 | - !ExpWarmup 11 | steps: 1000 12 | power: 4 13 | 14 | OptimizerBuilder: 15 | optimizer: 16 | momentum: 0.9 17 | type: Momentum 18 | regularizer: 19 | factor: 0.0001 20 | type: L2 21 | -------------------------------------------------------------------------------- /configs/mot/mtmct/README.md: -------------------------------------------------------------------------------- 1 | README_cn.md -------------------------------------------------------------------------------- /configs/mot/ocsort/README.md: -------------------------------------------------------------------------------- 1 | README_cn.md -------------------------------------------------------------------------------- /configs/mot/pedestrian/README.md: -------------------------------------------------------------------------------- 1 | README_cn.md -------------------------------------------------------------------------------- /configs/mot/vehicle/README.md: -------------------------------------------------------------------------------- 1 | README_cn.md -------------------------------------------------------------------------------- /configs/picodet/_base_/optimizer_300e.yml: -------------------------------------------------------------------------------- 1 | epoch: 300 2 | 3 | LearningRate: 4 | base_lr: 0.32 5 | schedulers: 6 | - name: CosineDecay 7 | max_epochs: 300 8 | - name: LinearWarmup 9 | start_factor: 0.1 10 | steps: 300 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | momentum: 0.9 15 | type: Momentum 16 | regularizer: 17 | factor: 0.00004 18 | type: L2 19 | -------------------------------------------------------------------------------- /configs/picodet/legacy_model/_base_/optimizer_100e.yml: -------------------------------------------------------------------------------- 1 | epoch: 100 2 | 3 | LearningRate: 4 | base_lr: 0.4 5 | schedulers: 6 | - name: CosineDecay 7 | max_epochs: 100 8 | - name: LinearWarmup 9 | start_factor: 0.1 10 | steps: 300 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | momentum: 0.9 15 | type: Momentum 16 | regularizer: 17 | factor: 0.00004 18 | type: L2 19 | -------------------------------------------------------------------------------- /configs/picodet/legacy_model/_base_/optimizer_300e.yml: -------------------------------------------------------------------------------- 1 | epoch: 300 2 | 3 | LearningRate: 4 | base_lr: 0.4 5 | schedulers: 6 | - name: CosineDecay 7 | max_epochs: 300 8 | - name: LinearWarmup 9 | start_factor: 0.1 10 | steps: 300 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | momentum: 0.9 15 | type: Momentum 16 | regularizer: 17 | factor: 0.00004 18 | type: L2 19 | -------------------------------------------------------------------------------- /configs/picodet/legacy_model/application/layout_analysis/images/layout_demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/picodet/legacy_model/application/layout_analysis/images/layout_demo.png -------------------------------------------------------------------------------- /configs/picodet/legacy_model/application/layout_analysis/images/layout_res.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/picodet/legacy_model/application/layout_analysis/images/layout_res.jpg -------------------------------------------------------------------------------- /configs/picodet/legacy_model/more_config/picodet_lcnet_1_5x_416_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../../datasets/coco_detection.yml', 3 | '../../../runtime.yml', 4 | '../_base_/picodet_esnet.yml', 5 | '../_base_/optimizer_300e.yml', 6 | '../_base_/picodet_416_reader.yml', 7 | ] 8 | 9 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/LCNet_x1_5_pretrained.pdparams 10 | weights: output/picodet_lcnet_1_5x_416_coco/model_final 11 | find_unused_parameters: True 12 | use_ema: true 13 | cycle_epoch: 40 14 | snapshot_epoch: 10 15 | 16 | PicoDet: 17 | backbone: LCNet 18 | neck: CSPPAN 19 | head: PicoHead 20 | 21 | LCNet: 22 | scale: 1.5 23 | feature_maps: [3, 4, 5] 24 | -------------------------------------------------------------------------------- /configs/picodet/legacy_model/picodet_m_320_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/coco_detection.yml', 3 | '../../runtime.yml', 4 | '_base_/picodet_esnet.yml', 5 | '_base_/optimizer_300e.yml', 6 | '_base_/picodet_320_reader.yml', 7 | ] 8 | 9 | weights: output/picodet_m_320_coco/model_final 10 | find_unused_parameters: True 11 | use_ema: true 12 | cycle_epoch: 40 13 | snapshot_epoch: 10 14 | -------------------------------------------------------------------------------- /configs/picodet/legacy_model/picodet_m_416_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/coco_detection.yml', 3 | '../../runtime.yml', 4 | '_base_/picodet_esnet.yml', 5 | '_base_/optimizer_300e.yml', 6 | '_base_/picodet_416_reader.yml', 7 | ] 8 | 9 | weights: output/picodet_m_416_coco/model_final 10 | find_unused_parameters: True 11 | use_ema: true 12 | cycle_epoch: 40 13 | snapshot_epoch: 10 14 | -------------------------------------------------------------------------------- /configs/picodet/legacy_model/pruner/optimizer_300e_pruner.yml: -------------------------------------------------------------------------------- 1 | epoch: 300 2 | 3 | LearningRate: 4 | base_lr: 0.15 5 | schedulers: 6 | - !CosineDecay 7 | max_epochs: 300 8 | - !LinearWarmup 9 | start_factor: 1.0 10 | steps: 34350 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | momentum: 0.9 15 | type: Momentum 16 | regularizer: 17 | factor: 0.00004 18 | type: L2 19 | -------------------------------------------------------------------------------- /configs/picodet/legacy_model/pruner/picodet_m_320_coco_pruner.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../../datasets/coco_detection.yml', 3 | '../../../runtime.yml', 4 | '../_base_/picodet_esnet.yml', 5 | './optimizer_300e_pruner.yml', 6 | '../_base_/picodet_320_reader.yml', 7 | ] 8 | 9 | weights: output/picodet_m_320_coco/model_final 10 | find_unused_parameters: True 11 | use_ema: true 12 | cycle_epoch: 40 13 | snapshot_epoch: 10 14 | -------------------------------------------------------------------------------- /configs/picodet/picodet_m_320_coco_lcnet.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/picodet_v2.yml', 5 | '_base_/optimizer_300e.yml', 6 | '_base_/picodet_320_reader.yml', 7 | ] 8 | 9 | weights: output/picodet_m_320_coco/best_model 10 | find_unused_parameters: True 11 | use_ema: true 12 | epoch: 300 13 | snapshot_epoch: 10 14 | 15 | TrainReader: 16 | batch_size: 48 17 | 18 | LearningRate: 19 | base_lr: 0.24 20 | schedulers: 21 | - name: CosineDecay 22 | max_epochs: 300 23 | - name: LinearWarmup 24 | start_factor: 0.1 25 | steps: 300 26 | -------------------------------------------------------------------------------- /configs/picodet/picodet_m_416_coco_lcnet.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/picodet_v2.yml', 5 | '_base_/optimizer_300e.yml', 6 | '_base_/picodet_416_reader.yml', 7 | ] 8 | 9 | weights: output/picodet_m_416_coco/best_model 10 | find_unused_parameters: True 11 | use_ema: true 12 | epoch: 250 13 | snapshot_epoch: 10 14 | 15 | TrainReader: 16 | batch_size: 48 17 | 18 | LearningRate: 19 | base_lr: 0.24 20 | schedulers: 21 | - name: CosineDecay 22 | max_epochs: 300 23 | - name: LinearWarmup 24 | start_factor: 0.1 25 | steps: 300 26 | -------------------------------------------------------------------------------- /configs/pphuman/pedestrian_yolov3/demo/001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/pphuman/pedestrian_yolov3/demo/001.png -------------------------------------------------------------------------------- /configs/pphuman/pedestrian_yolov3/demo/002.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/pphuman/pedestrian_yolov3/demo/002.png -------------------------------------------------------------------------------- /configs/pphuman/pedestrian_yolov3/demo/003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/pphuman/pedestrian_yolov3/demo/003.png -------------------------------------------------------------------------------- /configs/pphuman/pedestrian_yolov3/demo/004.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/pphuman/pedestrian_yolov3/demo/004.png -------------------------------------------------------------------------------- /configs/pphuman/pedestrian_yolov3/pedestrian.json: -------------------------------------------------------------------------------- 1 | { 2 | "images": [], 3 | "annotations": [], 4 | "categories": [ 5 | { 6 | "supercategory": "component", 7 | "id": 1, 8 | "name": "pedestrian" 9 | } 10 | ] 11 | } 12 | -------------------------------------------------------------------------------- /configs/ppvehicle/vehicle_yolov3/demo/001.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/ppvehicle/vehicle_yolov3/demo/001.jpeg -------------------------------------------------------------------------------- /configs/ppvehicle/vehicle_yolov3/demo/003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/ppvehicle/vehicle_yolov3/demo/003.png -------------------------------------------------------------------------------- /configs/ppvehicle/vehicle_yolov3/demo/004.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/ppvehicle/vehicle_yolov3/demo/004.png -------------------------------------------------------------------------------- /configs/ppvehicle/vehicle_yolov3/demo/005.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/ppvehicle/vehicle_yolov3/demo/005.png -------------------------------------------------------------------------------- /configs/ppyolo/_base_/optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 405 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: 9 | - 243 10 | - 324 11 | - !LinearWarmup 12 | start_factor: 0. 13 | steps: 4000 14 | 15 | OptimizerBuilder: 16 | clip_grad_by_norm: 35. 17 | optimizer: 18 | momentum: 0.9 19 | type: Momentum 20 | regularizer: 21 | factor: 0.0005 22 | type: L2 23 | -------------------------------------------------------------------------------- /configs/ppyolo/_base_/optimizer_2x.yml: -------------------------------------------------------------------------------- 1 | epoch: 811 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: 9 | - 649 10 | - 730 11 | - !LinearWarmup 12 | start_factor: 0. 13 | steps: 4000 14 | 15 | OptimizerBuilder: 16 | clip_grad_by_norm: 35. 17 | optimizer: 18 | momentum: 0.9 19 | type: Momentum 20 | regularizer: 21 | factor: 0.0005 22 | type: L2 23 | -------------------------------------------------------------------------------- /configs/ppyolo/_base_/optimizer_365e.yml: -------------------------------------------------------------------------------- 1 | epoch: 365 2 | 3 | LearningRate: 4 | base_lr: 0.005 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: 9 | - 243 10 | - !LinearWarmup 11 | start_factor: 0. 12 | steps: 4000 13 | 14 | OptimizerBuilder: 15 | clip_grad_by_norm: 35. 16 | optimizer: 17 | momentum: 0.9 18 | type: Momentum 19 | regularizer: 20 | factor: 0.0005 21 | type: L2 22 | -------------------------------------------------------------------------------- /configs/ppyolo/_base_/optimizer_650e.yml: -------------------------------------------------------------------------------- 1 | epoch: 650 2 | 3 | LearningRate: 4 | base_lr: 0.005 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: 9 | - 430 10 | - 540 11 | - 610 12 | - !LinearWarmup 13 | start_factor: 0. 14 | steps: 4000 15 | 16 | OptimizerBuilder: 17 | optimizer: 18 | momentum: 0.9 19 | type: Momentum 20 | regularizer: 21 | factor: 0.0005 22 | type: L2 23 | -------------------------------------------------------------------------------- /configs/ppyolo/ppyolo_r50vd_dcn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/ppyolo_r50vd_dcn.yml', 5 | './_base_/optimizer_1x.yml', 6 | './_base_/ppyolo_reader.yml', 7 | ] 8 | 9 | snapshot_epoch: 16 10 | weights: output/ppyolo_r50vd_dcn_1x_coco/model_final 11 | -------------------------------------------------------------------------------- /configs/ppyolo/ppyolo_r50vd_dcn_2x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/ppyolo_r50vd_dcn.yml', 5 | './_base_/optimizer_2x.yml', 6 | './_base_/ppyolo_reader.yml', 7 | ] 8 | 9 | snapshot_epoch: 16 10 | weights: output/ppyolo_r50vd_dcn_2x_coco/model_final 11 | -------------------------------------------------------------------------------- /configs/ppyolo/ppyolo_test.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/ppyolo_r50vd_dcn.yml', 5 | './_base_/optimizer_1x.yml', 6 | './_base_/ppyolo_reader.yml', 7 | ] 8 | 9 | snapshot_epoch: 16 10 | 11 | EvalDataset: 12 | !COCODataSet 13 | image_dir: test2017 14 | anno_path: annotations/image_info_test-dev2017.json 15 | dataset_dir: dataset/coco 16 | -------------------------------------------------------------------------------- /configs/ppyolo/ppyolo_tiny_650e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/ppyolo_tiny.yml', 5 | './_base_/optimizer_650e.yml', 6 | './_base_/ppyolo_tiny_reader.yml', 7 | ] 8 | 9 | snapshot_epoch: 1 10 | weights: output/ppyolo_tiny_650e_coco/model_final 11 | -------------------------------------------------------------------------------- /configs/ppyolo/ppyolov2_r101vd_dcn_365e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/ppyolov2_r50vd_dcn.yml', 5 | './_base_/optimizer_365e.yml', 6 | './_base_/ppyolov2_reader.yml', 7 | ] 8 | 9 | snapshot_epoch: 8 10 | weights: output/ppyolov2_r101vd_dcn_365e_coco/model_final 11 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet101_vd_ssld_pretrained.pdparams 12 | 13 | ResNet: 14 | depth: 101 15 | variant: d 16 | return_idx: [1, 2, 3] 17 | dcn_v2_stages: [3] 18 | freeze_at: -1 19 | freeze_norm: false 20 | norm_decay: 0. 21 | -------------------------------------------------------------------------------- /configs/ppyolo/ppyolov2_r50vd_dcn_365e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/ppyolov2_r50vd_dcn.yml', 5 | './_base_/optimizer_365e.yml', 6 | './_base_/ppyolov2_reader.yml', 7 | ] 8 | 9 | snapshot_epoch: 8 10 | weights: output/ppyolov2_r50vd_dcn_365e_coco/model_final 11 | -------------------------------------------------------------------------------- /configs/ppyoloe/_base_/optimizer_300e.yml: -------------------------------------------------------------------------------- 1 | epoch: 300 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - name: CosineDecay 7 | max_epochs: 360 8 | - name: LinearWarmup 9 | start_factor: 0. 10 | epochs: 5 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | momentum: 0.9 15 | type: Momentum 16 | regularizer: 17 | factor: 0.0005 18 | type: L2 19 | -------------------------------------------------------------------------------- /configs/ppyoloe/_base_/optimizer_36e_xpu.yml: -------------------------------------------------------------------------------- 1 | epoch: 36 2 | 3 | LearningRate: 4 | base_lr: 0.00125 5 | schedulers: 6 | - name: CosineDecay 7 | max_epochs: 43 8 | - name: LinearWarmup 9 | start_factor: 0.001 10 | steps: 2000 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | momentum: 0.9 15 | type: Momentum 16 | regularizer: 17 | factor: 0.0005 18 | type: L2 19 | -------------------------------------------------------------------------------- /configs/ppyoloe/_base_/optimizer_400e.yml: -------------------------------------------------------------------------------- 1 | epoch: 400 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - name: CosineDecay 7 | max_epochs: 480 8 | - name: LinearWarmup 9 | start_factor: 0. 10 | epochs: 5 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | momentum: 0.9 15 | type: Momentum 16 | regularizer: 17 | factor: 0.0005 18 | type: L2 19 | -------------------------------------------------------------------------------- /configs/ppyoloe/_base_/optimizer_60e.yml: -------------------------------------------------------------------------------- 1 | epoch: 60 2 | 3 | LearningRate: 4 | base_lr: 0.001 5 | schedulers: 6 | - name: CosineDecay 7 | max_epochs: 72 8 | - name: LinearWarmup 9 | start_factor: 0. 10 | epochs: 1 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | momentum: 0.9 15 | type: Momentum 16 | regularizer: 17 | factor: 0.0005 18 | type: L2 19 | -------------------------------------------------------------------------------- /configs/ppyoloe/_base_/optimizer_80e.yml: -------------------------------------------------------------------------------- 1 | epoch: 80 2 | 3 | LearningRate: 4 | base_lr: 0.001 5 | schedulers: 6 | - name: CosineDecay 7 | max_epochs: 96 8 | - name: LinearWarmup 9 | start_factor: 0. 10 | epochs: 5 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | momentum: 0.9 15 | type: Momentum 16 | regularizer: 17 | factor: 0.0005 18 | type: L2 19 | -------------------------------------------------------------------------------- /configs/ppyoloe/application/_base_/exdark_detection.yml: -------------------------------------------------------------------------------- 1 | metric: COCO 2 | num_classes: 12 3 | 4 | TrainDataset: 5 | !COCODataSet 6 | image_dir: images 7 | anno_path: coco_annotations/train.json 8 | dataset_dir: dataset/Exdark/ 9 | data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd'] 10 | 11 | EvalDataset: 12 | !COCODataSet 13 | image_dir: images 14 | anno_path: coco_annotations/val.json 15 | dataset_dir: dataset/Exdark/ 16 | 17 | TestDataset: 18 | !ImageFolder 19 | anno_path: coco_annotations/val.json # also support txt (like VOC's label_list.txt) 20 | dataset_dir: dataset/Exdark/ # if set, anno_path will be 'dataset_dir/anno_path' 21 | -------------------------------------------------------------------------------- /configs/ppyoloe/application/_base_/pcb_detection.yml: -------------------------------------------------------------------------------- 1 | metric: COCO 2 | num_classes: 6 3 | 4 | TrainDataset: 5 | !COCODataSet 6 | image_dir: images 7 | anno_path: pcb_cocoanno/train.json 8 | dataset_dir: dataset/PCB_coco/ 9 | data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd'] 10 | 11 | EvalDataset: 12 | !COCODataSet 13 | image_dir: images 14 | anno_path: pcb_cocoanno/val.json 15 | dataset_dir: dataset/PCB_coco/ 16 | 17 | TestDataset: 18 | !ImageFolder 19 | anno_path: pcb_cocoanno/val.json # also support txt (like VOC's label_list.txt) 20 | dataset_dir: dataset/PCB_coco/ # if set, anno_path will be 'dataset_dir/anno_path' 21 | -------------------------------------------------------------------------------- /configs/ppyoloe/application/_base_/sku110k.yml: -------------------------------------------------------------------------------- 1 | metric: COCO 2 | num_classes: 1 3 | 4 | TrainDataset: 5 | !COCODataSet 6 | image_dir: images 7 | anno_path: annotations/annotations_train.json 8 | dataset_dir: dataset/SKU110K_fixed 9 | data_fields: ['image', 'gt_bbox', 'gt_class', 'difficult'] 10 | 11 | EvalDataset: 12 | !COCODataSet 13 | image_dir: images 14 | anno_path: annotations/annotations_val.json 15 | dataset_dir: dataset/SKU110K_fixed 16 | allow_empty: true 17 | 18 | TestDataset: 19 | !ImageFolder 20 | anno_path: annotations/annotations_test.json 21 | dataset_dir: dataset/SKU110K_fixed 22 | -------------------------------------------------------------------------------- /configs/ppyoloe/application/_base_/wgisd_detection.yml: -------------------------------------------------------------------------------- 1 | metric: COCO 2 | num_classes: 5 3 | 4 | TrainDataset: 5 | !COCODataSet 6 | image_dir: data 7 | anno_path: coco_annotations/new_train_bbox_instances.json 8 | dataset_dir: dataset/wgisd/ 9 | data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd'] 10 | 11 | EvalDataset: 12 | !COCODataSet 13 | image_dir: data 14 | anno_path: coco_annotations/new_test_bbox_instances.json 15 | dataset_dir: dataset/wgisd/ 16 | 17 | TestDataset: 18 | !ImageFolder 19 | anno_path: coco_annotations/new_test_bbox_instances.json # also support txt (like VOC's label_list.txt) 20 | dataset_dir: dataset/wgisd/ # if set, anno_path will be 'dataset_dir/anno_path' 21 | -------------------------------------------------------------------------------- /configs/ppyoloe/application/ppyoloe_crn_m_80e_exdark.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | './_base_/exdark_detection.yml', 3 | '../../runtime.yml', 4 | '../_base_/optimizer_80e.yml', 5 | '../_base_/ppyoloe_crn.yml', 6 | '../_base_/ppyoloe_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_crn_m_80e_exdark/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/ppyoloe_crn_m_300e_coco.pdparams 14 | depth_mult: 0.67 15 | width_mult: 0.75 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/application/ppyoloe_crn_m_80e_pcb.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | './_base_/pcb_detection.yml', 3 | '../../runtime.yml', 4 | '../_base_/optimizer_80e.yml', 5 | '../_base_/ppyoloe_crn.yml', 6 | '../_base_/ppyoloe_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_crn_m_80e_pcb/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/ppyoloe_crn_m_300e_coco.pdparams 14 | depth_mult: 0.67 15 | width_mult: 0.75 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/application/ppyoloe_crn_m_80e_wgisd.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | './_base_/wgisd_detection.yml', 3 | '../../runtime.yml', 4 | '../_base_/optimizer_80e.yml', 5 | '../_base_/ppyoloe_crn.yml', 6 | '../_base_/ppyoloe_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_crn_m_80e_wgisd/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/ppyoloe_crn_m_300e_coco.pdparams 14 | depth_mult: 0.67 15 | width_mult: 0.75 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/application/ppyoloe_plus_crn_m_80e_coco_pretrained_exdark.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | './_base_/exdark_detection.yml', 3 | '../../runtime.yml', 4 | '../_base_/optimizer_80e.yml', 5 | '../_base_/ppyoloe_plus_crn.yml', 6 | '../_base_/ppyoloe_plus_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_plus_crn_m_80e_coco_pretrained_exdark/model_final 12 | 13 | pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/ppyoloe_plus_crn_m_80e_coco.pdparams 14 | depth_mult: 0.67 15 | width_mult: 0.75 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/application/ppyoloe_plus_crn_m_80e_coco_pretrained_pcb.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | './_base_/pcb_detection.yml', 3 | '../../runtime.yml', 4 | '../_base_/optimizer_80e.yml', 5 | '../_base_/ppyoloe_plus_crn.yml', 6 | '../_base_/ppyoloe_plus_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_plus_crn_m_80e_coco_pretrained_pcb/model_final 12 | 13 | pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/ppyoloe_plus_crn_m_80e_coco.pdparams 14 | depth_mult: 0.67 15 | width_mult: 0.75 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/application/ppyoloe_plus_crn_m_80e_coco_pretrained_wgisd.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | './_base_/wgisd_detection.yml', 3 | '../../runtime.yml', 4 | '../_base_/optimizer_80e.yml', 5 | '../_base_/ppyoloe_plus_crn.yml', 6 | '../_base_/ppyoloe_plus_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_plus_crn_m_80e_coco_pretrained_wgisd/model_final 12 | 13 | pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/ppyoloe_plus_crn_m_80e_coco.pdparams 14 | depth_mult: 0.67 15 | width_mult: 0.75 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/application/ppyoloe_plus_crn_m_80e_obj365_pretrained_exdark.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | './_base_/exdark_detection.yml', 3 | '../../runtime.yml', 4 | '../_base_/optimizer_80e.yml', 5 | '../_base_/ppyoloe_plus_crn.yml', 6 | '../_base_/ppyoloe_plus_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_plus_crn_m_80e_obj365_pretrained_exdark/model_final 12 | 13 | pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_m_obj365_pretrained.pdparams 14 | depth_mult: 0.67 15 | width_mult: 0.75 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/application/ppyoloe_plus_crn_m_80e_obj365_pretrained_pcb.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | './_base_/pcb_detection.yml', 3 | '../../runtime.yml', 4 | '../_base_/optimizer_80e.yml', 5 | '../_base_/ppyoloe_plus_crn.yml', 6 | '../_base_/ppyoloe_plus_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_plus_crn_m_80e_obj365_pretrained_pcb/model_final 12 | 13 | pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_m_obj365_pretrained.pdparams 14 | depth_mult: 0.67 15 | width_mult: 0.75 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/application/ppyoloe_plus_crn_m_80e_obj365_pretrained_wgisd.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | './_base_/wgisd_detection.yml', 3 | '../../runtime.yml', 4 | '../_base_/optimizer_80e.yml', 5 | '../_base_/ppyoloe_plus_crn.yml', 6 | '../_base_/ppyoloe_plus_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_plus_crn_m_80e_obj365_pretrained_wgisd/model_final 12 | 13 | pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_m_obj365_pretrained.pdparams 14 | depth_mult: 0.67 15 | width_mult: 0.75 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/objects365/ppyoloe_plus_crn_l_60e_objects365.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/objects365_detection.yml', 3 | '../../runtime.yml', 4 | '../_base_/optimizer_60e.yml', 5 | '../_base_/ppyoloe_plus_crn.yml', 6 | '../_base_/ppyoloe_plus_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_plus_crn_l_60e_objects365/model_final 12 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_l_pretrained.pdparams 13 | 14 | CSPResNet: 15 | use_alpha: False 16 | 17 | PPYOLOEHead: 18 | static_assigner_epoch: 20 19 | 20 | depth_mult: 1.0 21 | width_mult: 1.0 22 | -------------------------------------------------------------------------------- /configs/ppyoloe/objects365/ppyoloe_plus_crn_m_60e_objects365.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/objects365_detection.yml', 3 | '../../runtime.yml', 4 | '../_base_/optimizer_60e.yml', 5 | '../_base_/ppyoloe_plus_crn.yml', 6 | '../_base_/ppyoloe_plus_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_plus_crn_m_60e_objects365/model_final 12 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_m_pretrained.pdparams 13 | 14 | CSPResNet: 15 | use_alpha: False 16 | 17 | PPYOLOEHead: 18 | static_assigner_epoch: 20 19 | 20 | depth_mult: 0.67 21 | width_mult: 0.75 22 | -------------------------------------------------------------------------------- /configs/ppyoloe/objects365/ppyoloe_plus_crn_s_60e_objects365.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/objects365_detection.yml', 3 | '../../runtime.yml', 4 | '../_base_/optimizer_60e.yml', 5 | '../_base_/ppyoloe_plus_crn.yml', 6 | '../_base_/ppyoloe_plus_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_plus_crn_s_60e_objects365/model_final 12 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_s_pretrained.pdparams 13 | 14 | CSPResNet: 15 | use_alpha: False 16 | 17 | PPYOLOEHead: 18 | static_assigner_epoch: 20 19 | 20 | depth_mult: 0.33 21 | width_mult: 0.50 22 | -------------------------------------------------------------------------------- /configs/ppyoloe/objects365/ppyoloe_plus_crn_x_60e_objects365.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/objects365_detection.yml', 3 | '../../runtime.yml', 4 | '../_base_/optimizer_60e.yml', 5 | '../_base_/ppyoloe_plus_crn.yml', 6 | '../_base_/ppyoloe_plus_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_plus_crn_x_60e_objects365/model_final 12 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_x_pretrained.pdparams 13 | 14 | CSPResNet: 15 | use_alpha: False 16 | 17 | PPYOLOEHead: 18 | static_assigner_epoch: 20 19 | 20 | depth_mult: 1.33 21 | width_mult: 1.25 22 | -------------------------------------------------------------------------------- /configs/ppyoloe/ppyoloe_crn_l_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/ppyoloe_crn.yml', 6 | './_base_/ppyoloe_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 10 11 | weights: output/ppyoloe_crn_l_300e_coco/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_l_pretrained.pdparams 14 | depth_mult: 1.0 15 | width_mult: 1.0 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/ppyoloe_crn_m_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/ppyoloe_crn.yml', 6 | './_base_/ppyoloe_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 10 11 | weights: output/ppyoloe_crn_m_300e_coco/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_m_pretrained.pdparams 14 | depth_mult: 0.67 15 | width_mult: 0.75 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/ppyoloe_crn_s_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/ppyoloe_crn.yml', 6 | './_base_/ppyoloe_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 10 11 | weights: output/ppyoloe_crn_s_300e_coco/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_s_pretrained.pdparams 14 | depth_mult: 0.33 15 | width_mult: 0.50 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/ppyoloe_crn_s_400e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_400e.yml', 5 | './_base_/ppyoloe_crn.yml', 6 | './_base_/ppyoloe_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 10 11 | weights: output/ppyoloe_crn_s_400e_coco/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_s_pretrained.pdparams 14 | depth_mult: 0.33 15 | width_mult: 0.50 16 | 17 | PPYOLOEHead: 18 | static_assigner_epoch: 133 19 | -------------------------------------------------------------------------------- /configs/ppyoloe/ppyoloe_crn_x_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/ppyoloe_crn.yml', 6 | './_base_/ppyoloe_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 10 11 | weights: output/ppyoloe_crn_x_300e_coco/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_x_pretrained.pdparams 14 | depth_mult: 1.33 15 | width_mult: 1.25 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/ppyoloe_plus_crn_l_80e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_80e.yml', 5 | './_base_/ppyoloe_plus_crn.yml', 6 | './_base_/ppyoloe_plus_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_plus_crn_l_80e_coco/model_final 12 | 13 | pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_l_obj365_pretrained.pdparams 14 | depth_mult: 1.0 15 | width_mult: 1.0 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/ppyoloe_plus_crn_m_80e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_80e.yml', 5 | './_base_/ppyoloe_plus_crn.yml', 6 | './_base_/ppyoloe_plus_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_plus_crn_m_80e_coco/model_final 12 | 13 | pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_m_obj365_pretrained.pdparams 14 | depth_mult: 0.67 15 | width_mult: 0.75 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/ppyoloe_plus_crn_s_80e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_80e.yml', 5 | './_base_/ppyoloe_plus_crn.yml', 6 | './_base_/ppyoloe_plus_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_plus_crn_s_80e_coco/model_final 12 | 13 | pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_s_obj365_pretrained.pdparams 14 | depth_mult: 0.33 15 | width_mult: 0.50 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/ppyoloe_plus_crn_t_auxhead_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/ppyoloe_plus_crn_tiny_auxhead.yml', 6 | './_base_/ppyoloe_plus_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 10 11 | weights: output/ppyoloe_plus_crn_t_auxhead_300e_coco/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_t_pretrained.pdparams 14 | depth_mult: 0.33 15 | width_mult: 0.375 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/ppyoloe_plus_crn_t_auxhead_320_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/ppyoloe_plus_crn_tiny_auxhead.yml', 6 | './_base_/ppyoloe_plus_reader_320.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 10 11 | weights: output/ppyoloe_plus_crn_t_auxhead_320_300e_coco/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_t_pretrained.pdparams 14 | depth_mult: 0.33 15 | width_mult: 0.375 16 | -------------------------------------------------------------------------------- /configs/ppyoloe/ppyoloe_plus_crn_t_auxhead_relu_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/ppyoloe_plus_crn_tiny_auxhead.yml', 6 | './_base_/ppyoloe_plus_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 10 11 | weights: output/ppyoloe_plus_crn_t_auxhead_relu_300e_coco/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_t_pretrained.pdparams 14 | depth_mult: 0.33 15 | width_mult: 0.375 16 | 17 | 18 | CSPResNet: 19 | act: 'relu' 20 | 21 | CustomCSPPAN: 22 | act: 'relu' 23 | 24 | PPYOLOEHead: 25 | act: 'relu' 26 | attn_conv: None 27 | -------------------------------------------------------------------------------- /configs/ppyoloe/ppyoloe_plus_crn_t_auxhead_relu_320_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/ppyoloe_plus_crn_tiny_auxhead.yml', 6 | './_base_/ppyoloe_plus_reader_320.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 10 11 | weights: output/ppyoloe_plus_crn_t_auxhead_relu_320_300e_coco/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_t_pretrained.pdparams 14 | depth_mult: 0.33 15 | width_mult: 0.375 16 | 17 | 18 | CSPResNet: 19 | act: 'relu' 20 | 21 | CustomCSPPAN: 22 | act: 'relu' 23 | 24 | PPYOLOEHead: 25 | act: 'relu' 26 | attn_conv: None 27 | -------------------------------------------------------------------------------- /configs/ppyoloe/ppyoloe_plus_crn_x_80e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_80e.yml', 5 | './_base_/ppyoloe_plus_crn.yml', 6 | './_base_/ppyoloe_plus_reader.yml', 7 | ] 8 | 9 | log_iter: 100 10 | snapshot_epoch: 5 11 | weights: output/ppyoloe_plus_crn_x_80e_coco/model_final 12 | 13 | pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/pretrained/ppyoloe_crn_x_obj365_pretrained.pdparams 14 | depth_mult: 1.33 15 | width_mult: 1.25 16 | -------------------------------------------------------------------------------- /configs/queryinst/_base_/optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.0001 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [8, 11] 9 | - !LinearWarmup 10 | start_factor: 0.001 11 | steps: 1000 12 | 13 | OptimizerBuilder: 14 | clip_grad_by_norm: 0.1 15 | optimizer: 16 | type: AdamW 17 | weight_decay: 0.0001 18 | -------------------------------------------------------------------------------- /configs/queryinst/queryinst_r50_fpn_1x_pro100_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_instance.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1x.yml', 5 | '_base_/queryinst_r50_fpn.yml', 6 | '_base_/queryinst_reader.yml', 7 | ] 8 | 9 | log_iter: 50 10 | find_unused_parameters: true 11 | 12 | weights: output/queryinst_r50_fpn_1x_pro100_coco/model_final 13 | -------------------------------------------------------------------------------- /configs/rcnn_enhance/_base_/optimizer_3x.yml: -------------------------------------------------------------------------------- 1 | epoch: 36 2 | 3 | LearningRate: 4 | base_lr: 0.02 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [24, 33] 9 | - !LinearWarmup 10 | start_factor: 0. 11 | steps: 1000 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0001 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/rcnn_enhance/faster_rcnn_enhance_3x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_3x.yml', 5 | '_base_/faster_rcnn_enhance.yml', 6 | '_base_/faster_rcnn_enhance_reader.yml', 7 | ] 8 | weights: output/faster_rcnn_enhance_r50_3x_coco/model_final 9 | -------------------------------------------------------------------------------- /configs/retinanet/_base_/optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [8, 11] 9 | - !LinearWarmup 10 | start_factor: 0.001 11 | steps: 500 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0001 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/retinanet/_base_/optimizer_2x.yml: -------------------------------------------------------------------------------- 1 | epoch: 24 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [16, 22] 9 | - !LinearWarmup 10 | start_factor: 0.001 11 | steps: 500 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0001 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/retinanet/retinanet_r101_distill_r50_2x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/retinanet_r50_fpn.yml', 5 | '_base_/optimizer_2x.yml', 6 | '_base_/retinanet_reader.yml' 7 | ] 8 | 9 | weights: https://paddledet.bj.bcebos.com/models/retinanet_r101_distill_r50_2x_coco.pdparams 10 | -------------------------------------------------------------------------------- /configs/retinanet/retinanet_r101_fpn_2x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/retinanet_r50_fpn.yml', 5 | '_base_/optimizer_2x.yml', 6 | '_base_/retinanet_reader.yml' 7 | ] 8 | 9 | weights: output/retinanet_r101_fpn_2x_coco/model_final 10 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet101_pretrained.pdparams 11 | 12 | ResNet: 13 | depth: 101 14 | variant: b 15 | norm_type: bn 16 | freeze_at: 0 17 | return_idx: [1, 2, 3] 18 | num_stages: 4 19 | -------------------------------------------------------------------------------- /configs/retinanet/retinanet_r50_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/retinanet_r50_fpn.yml', 5 | '_base_/optimizer_1x.yml', 6 | '_base_/retinanet_reader.yml' 7 | ] 8 | 9 | weights: output/retinanet_r50_fpn_1x_coco/model_final 10 | -------------------------------------------------------------------------------- /configs/retinanet/retinanet_r50_fpn_2x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/retinanet_r50_fpn.yml', 5 | '_base_/optimizer_2x.yml', 6 | '_base_/retinanet_reader.yml' 7 | ] 8 | 9 | weights: output/retinanet_r50_fpn_2x_coco/model_final 10 | -------------------------------------------------------------------------------- /configs/rotate/fcosr/_base_/optimizer_3x.yml: -------------------------------------------------------------------------------- 1 | epoch: 36 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [24, 33] 9 | - !LinearWarmup 10 | start_factor: 0.3333333 11 | steps: 500 12 | 13 | OptimizerBuilder: 14 | clip_grad_by_norm: 35. 15 | optimizer: 16 | momentum: 0.9 17 | type: Momentum 18 | regularizer: 19 | factor: 0.0001 20 | type: L2 21 | -------------------------------------------------------------------------------- /configs/rotate/fcosr/fcosr_x50_3x_dota.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/dota.yml', 3 | '../../runtime.yml', 4 | '_base_/optimizer_3x.yml', 5 | '_base_/fcosr_reader.yml', 6 | '_base_/fcosr_x50.yml' 7 | ] 8 | 9 | weights: output/fcosr_x50_3x_dota/model_final 10 | -------------------------------------------------------------------------------- /configs/rotate/ppyoloe_r/_base_/optimizer_3x.yml: -------------------------------------------------------------------------------- 1 | epoch: 36 2 | 3 | LearningRate: 4 | base_lr: 0.008 5 | schedulers: 6 | - !CosineDecay 7 | max_epochs: 44 8 | - !LinearWarmup 9 | start_factor: 0. 10 | steps: 1000 11 | 12 | OptimizerBuilder: 13 | clip_grad_by_norm: 35. 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0005 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/rotate/ppyoloe_r/ppyoloe_r_crn_l_3x_dota.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/dota.yml', 3 | '../../runtime.yml', 4 | '_base_/optimizer_3x.yml', 5 | '_base_/ppyoloe_r_reader.yml', 6 | '_base_/ppyoloe_r_crn.yml' 7 | ] 8 | 9 | log_iter: 50 10 | snapshot_epoch: 1 11 | weights: output/ppyoloe_r_crn_l_3x_dota/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_l_pretrained.pdparams 14 | depth_mult: 1.0 15 | width_mult: 1.0 16 | -------------------------------------------------------------------------------- /configs/rotate/ppyoloe_r/ppyoloe_r_crn_l_3x_dota_ms.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/dota_ms.yml', 3 | '../../runtime.yml', 4 | '_base_/optimizer_3x.yml', 5 | '_base_/ppyoloe_r_reader.yml', 6 | '_base_/ppyoloe_r_crn.yml' 7 | ] 8 | 9 | log_iter: 50 10 | snapshot_epoch: 1 11 | weights: output/ppyoloe_r_crn_l_3x_dota/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_l_pretrained.pdparams 14 | depth_mult: 1.0 15 | width_mult: 1.0 16 | -------------------------------------------------------------------------------- /configs/rotate/ppyoloe_r/ppyoloe_r_crn_m_3x_dota.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/dota.yml', 3 | '../../runtime.yml', 4 | '_base_/optimizer_3x.yml', 5 | '_base_/ppyoloe_r_reader.yml', 6 | '_base_/ppyoloe_r_crn.yml' 7 | ] 8 | 9 | log_iter: 50 10 | snapshot_epoch: 1 11 | weights: output/ppyoloe_r_crn_m_3x_dota/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_m_pretrained.pdparams 14 | depth_mult: 0.67 15 | width_mult: 0.75 16 | -------------------------------------------------------------------------------- /configs/rotate/ppyoloe_r/ppyoloe_r_crn_m_3x_dota_ms.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/dota_ms.yml', 3 | '../../runtime.yml', 4 | '_base_/optimizer_3x.yml', 5 | '_base_/ppyoloe_r_reader.yml', 6 | '_base_/ppyoloe_r_crn.yml' 7 | ] 8 | 9 | log_iter: 50 10 | snapshot_epoch: 1 11 | weights: output/ppyoloe_r_crn_m_3x_dota/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_m_pretrained.pdparams 14 | depth_mult: 0.67 15 | width_mult: 0.75 16 | -------------------------------------------------------------------------------- /configs/rotate/ppyoloe_r/ppyoloe_r_crn_s_3x_dota.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/dota.yml', 3 | '../../runtime.yml', 4 | '_base_/optimizer_3x.yml', 5 | '_base_/ppyoloe_r_reader.yml', 6 | '_base_/ppyoloe_r_crn.yml' 7 | ] 8 | 9 | log_iter: 50 10 | snapshot_epoch: 1 11 | weights: output/ppyoloe_r_crn_s_3x_dota/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_s_pretrained.pdparams 14 | depth_mult: 0.33 15 | width_mult: 0.50 16 | -------------------------------------------------------------------------------- /configs/rotate/ppyoloe_r/ppyoloe_r_crn_s_3x_dota_ms.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/dota_ms.yml', 3 | '../../runtime.yml', 4 | '_base_/optimizer_3x.yml', 5 | '_base_/ppyoloe_r_reader.yml', 6 | '_base_/ppyoloe_r_crn.yml' 7 | ] 8 | 9 | log_iter: 50 10 | snapshot_epoch: 1 11 | weights: output/ppyoloe_r_crn_s_3x_dota/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_s_pretrained.pdparams 14 | depth_mult: 0.33 15 | width_mult: 0.50 16 | -------------------------------------------------------------------------------- /configs/rotate/ppyoloe_r/ppyoloe_r_crn_x_3x_dota.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/dota.yml', 3 | '../../runtime.yml', 4 | '_base_/optimizer_3x.yml', 5 | '_base_/ppyoloe_r_reader.yml', 6 | '_base_/ppyoloe_r_crn.yml' 7 | ] 8 | 9 | log_iter: 50 10 | snapshot_epoch: 1 11 | weights: output/ppyoloe_r_crn_x_3x_dota/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_x_pretrained.pdparams 14 | depth_mult: 1.33 15 | width_mult: 1.25 16 | -------------------------------------------------------------------------------- /configs/rotate/ppyoloe_r/ppyoloe_r_crn_x_3x_dota_ms.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/dota_ms.yml', 3 | '../../runtime.yml', 4 | '_base_/optimizer_3x.yml', 5 | '_base_/ppyoloe_r_reader.yml', 6 | '_base_/ppyoloe_r_crn.yml' 7 | ] 8 | 9 | log_iter: 50 10 | snapshot_epoch: 1 11 | weights: output/ppyoloe_r_crn_x_3x_dota/model_final 12 | 13 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_x_pretrained.pdparams 14 | depth_mult: 1.33 15 | width_mult: 1.25 16 | -------------------------------------------------------------------------------- /configs/rotate/s2anet/_base_/s2anet_optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [7, 10] 9 | - !LinearWarmup 10 | start_factor: 0.3333333333333333 11 | steps: 500 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0001 19 | type: L2 20 | clip_grad_by_norm: 35 21 | -------------------------------------------------------------------------------- /configs/rotate/s2anet/_base_/s2anet_optimizer_2x.yml: -------------------------------------------------------------------------------- 1 | epoch: 24 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [14, 20] 9 | - !LinearWarmup 10 | start_factor: 0.3333333333333333 11 | steps: 1000 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0001 19 | type: L2 20 | clip_grad_by_norm: 35 21 | -------------------------------------------------------------------------------- /configs/rotate/s2anet/s2anet_1x_spine.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/spine_coco.yml', 3 | '../../runtime.yml', 4 | '_base_/s2anet_optimizer_1x.yml', 5 | '_base_/s2anet.yml', 6 | '_base_/s2anet_reader.yml', 7 | ] 8 | 9 | weights: output/s2anet_1x_spine/model_final 10 | pretrain_weights: https://paddledet.bj.bcebos.com/models/s2anet_alignconv_2x_dota.pdparams 11 | 12 | # for 4 card 13 | LearningRate: 14 | base_lr: 0.01 15 | schedulers: 16 | - !PiecewiseDecay 17 | gamma: 0.1 18 | milestones: [7, 10] 19 | - !LinearWarmup 20 | start_factor: 0.3333333333333333 21 | epochs: 5 22 | 23 | S2ANetHead: 24 | reg_loss_weight: [1.0, 1.0, 1.0, 1.0, 1.05] 25 | cls_loss_weight: [1.05, 1.0] 26 | -------------------------------------------------------------------------------- /configs/rotate/s2anet/s2anet_alignconv_2x_dota.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/dota.yml', 3 | '../../runtime.yml', 4 | '_base_/s2anet_optimizer_2x.yml', 5 | '_base_/s2anet.yml', 6 | '_base_/s2anet_reader.yml', 7 | ] 8 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_ssld_v2_pretrained.pdparams 9 | 10 | weights: output/s2anet_alignconv_2x_dota/model_final 11 | -------------------------------------------------------------------------------- /configs/rotate/s2anet/s2anet_conv_2x_dota.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/dota.yml', 3 | '../../runtime.yml', 4 | '_base_/s2anet_optimizer_2x.yml', 5 | '_base_/s2anet.yml', 6 | '_base_/s2anet_reader.yml', 7 | ] 8 | weights: output/s2anet_conv_1x_dota/model_final 9 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_cos_pretrained.pdparams 10 | 11 | ResNet: 12 | depth: 50 13 | variant: b 14 | norm_type: bn 15 | return_idx: [1,2,3] 16 | num_stages: 4 17 | 18 | S2ANetHead: 19 | align_conv_type: 'Conv' 20 | -------------------------------------------------------------------------------- /configs/rtdetr/_base_/optimizer_6x.yml: -------------------------------------------------------------------------------- 1 | epoch: 72 2 | 3 | LearningRate: 4 | base_lr: 0.0001 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 1.0 8 | milestones: [100] 9 | use_warmup: true 10 | - !LinearWarmup 11 | start_factor: 0.001 12 | steps: 2000 13 | 14 | OptimizerBuilder: 15 | clip_grad_by_norm: 0.1 16 | regularizer: false 17 | optimizer: 18 | type: AdamW 19 | weight_decay: 0.0001 20 | -------------------------------------------------------------------------------- /configs/rtdetr/rtdetr_hgnetv2_l_6x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_6x.yml', 5 | '_base_/rtdetr_r50vd.yml', 6 | '_base_/rtdetr_reader.yml', 7 | ] 8 | 9 | weights: output/rtdetr_hgnetv2_l_6x_coco/model_final 10 | pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/pretrained/PPHGNetV2_L_ssld_pretrained.pdparams 11 | find_unused_parameters: True 12 | log_iter: 200 13 | 14 | 15 | DETR: 16 | backbone: PPHGNetV2 17 | 18 | PPHGNetV2: 19 | arch: 'L' 20 | return_idx: [1, 2, 3] 21 | freeze_stem_only: True 22 | freeze_at: 0 23 | freeze_norm: True 24 | lr_mult_list: [0., 0.05, 0.05, 0.05, 0.05] 25 | -------------------------------------------------------------------------------- /configs/rtdetr/rtdetr_r50vd_6x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_6x.yml', 5 | '_base_/rtdetr_r50vd.yml', 6 | '_base_/rtdetr_reader.yml', 7 | ] 8 | 9 | weights: output/rtdetr_r50vd_6x_coco/model_final 10 | find_unused_parameters: True 11 | log_iter: 200 12 | -------------------------------------------------------------------------------- /configs/rtdetr/rtdetr_r50vd_m_6x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_6x.yml', 5 | '_base_/rtdetr_r50vd.yml', 6 | '_base_/rtdetr_reader.yml', 7 | ] 8 | 9 | weights: output/rtdetr_r50vd_m_6x_coco/model_final 10 | find_unused_parameters: True 11 | log_iter: 200 12 | 13 | HybridEncoder: 14 | hidden_dim: 256 15 | use_encoder_idx: [2] 16 | num_encoder_layers: 1 17 | encoder_layer: 18 | name: TransformerLayer 19 | d_model: 256 20 | nhead: 8 21 | dim_feedforward: 1024 22 | dropout: 0. 23 | activation: 'gelu' 24 | expansion: 0.5 25 | depth_mult: 1.0 26 | 27 | RTDETRTransformer: 28 | eval_idx: 2 # use 3th decoder layer to eval 29 | -------------------------------------------------------------------------------- /configs/rtmdet/_base_/optimizer_300e.yml: -------------------------------------------------------------------------------- 1 | # set the same as YOLOX scheduler temporarily 2 | epoch: 300 3 | 4 | LearningRate: 5 | base_lr: 0.0004 6 | schedulers: 7 | - !CosineDecay 8 | max_epochs: 300 9 | - !LinearWarmup 10 | start_factor: 0.00001 11 | steps: 1000 12 | 13 | OptimizerBuilder: 14 | regularizer: false 15 | optimizer: 16 | type: AdamW 17 | weight_decay: 0.05 18 | -------------------------------------------------------------------------------- /configs/rtmdet/_base_/rtmdet_cspnext.yml: -------------------------------------------------------------------------------- 1 | architecture: RTMDet 2 | norm_type: sync_bn 3 | use_ema: True 4 | ema_decay: 0.9998 5 | ema_decay_type: "exponential" 6 | act: silu 7 | find_unused_parameters: True 8 | 9 | depth_mult: 1.0 10 | width_mult: 1.0 11 | 12 | RTMDet: 13 | backbone: CSPNeXt 14 | neck: CSPNeXtPAFPN 15 | head: RTMDetHead 16 | 17 | CSPNeXt: 18 | arch: "P5" 19 | return_idx: [2, 3, 4] 20 | 21 | # use default config 22 | # CSPNeXtPAFPN: 23 | 24 | RTMDetHead: 25 | exp_on_reg: False 26 | fpn_strides: [8, 16, 32] 27 | grid_cell_offset: 0 28 | nms: 29 | name: MultiClassNMS 30 | nms_top_k: 1000 31 | keep_top_k: 100 32 | score_threshold: 0.05 33 | nms_threshold: 0.6 34 | -------------------------------------------------------------------------------- /configs/rtmdet/rtmdet_l_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/rtmdet_cspnext.yml', 6 | './_base_/rtmdet_reader.yml', 7 | ] 8 | depth_mult: 1.0 9 | width_mult: 1.0 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/rtmnet_l_300e_coco/model_final 14 | 15 | RTMDetHead: 16 | exp_on_reg: True 17 | -------------------------------------------------------------------------------- /configs/rtmdet/rtmdet_m_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/rtmdet_cspnext.yml', 6 | './_base_/rtmdet_reader.yml', 7 | ] 8 | depth_mult: 0.67 9 | width_mult: 0.75 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/rtmnet_m_300e_coco/model_final 14 | 15 | RTMDetHead: 16 | exp_on_reg: True 17 | -------------------------------------------------------------------------------- /configs/rtmdet/rtmdet_s_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/rtmdet_cspnext.yml', 6 | './_base_/rtmdet_reader.yml', 7 | ] 8 | depth_mult: 0.33 9 | width_mult: 0.50 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/rtmnet_s_300e_coco/model_final 14 | pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/pretrained/cspnext_s_pretrained.pdparams 15 | -------------------------------------------------------------------------------- /configs/rtmdet/rtmdet_t_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/rtmdet_cspnext.yml', 6 | './_base_/rtmdet_reader.yml', 7 | ] 8 | depth_mult: 0.167 # 0.33 in yolox-tiny 9 | width_mult: 0.375 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/rtmnet_t_300e_coco/model_final 14 | pretrain_weights: https://bj.bcebos.com/v1/paddledet/models/pretrained/cspnext_t_pretrained.pdparams 15 | -------------------------------------------------------------------------------- /configs/rtmdet/rtmdet_x_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/rtmdet_cspnext.yml', 6 | './_base_/rtmdet_reader.yml', 7 | ] 8 | depth_mult: 1.33 9 | width_mult: 1.25 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/rtmnet_x_300e_coco/model_final 14 | 15 | RTMDetHead: 16 | exp_on_reg: True 17 | -------------------------------------------------------------------------------- /configs/runtime.yml: -------------------------------------------------------------------------------- 1 | use_gpu: true 2 | use_xpu: false 3 | use_mlu: false 4 | use_npu: false 5 | log_iter: 20 6 | save_dir: output 7 | snapshot_epoch: 1 8 | print_flops: false 9 | print_params: false 10 | 11 | # Exporting the model 12 | export: 13 | post_process: True # Whether post-processing is included in the network when export model. 14 | nms: True # Whether NMS is included in the network when export model. 15 | benchmark: False # It is used to testing model performance, if set `True`, post-process and NMS will not be exported. 16 | fuse_conv_bn: False 17 | -------------------------------------------------------------------------------- /configs/semi_det/arsl/arsl_fcos_r50_fpn_coco_full.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../_base_/coco_detection_full.yml', 3 | '../../runtime.yml', 4 | '_base_/arsl_fcos_r50_fpn.yml', 5 | '_base_/optimizer_360k.yml', 6 | '_base_/arsl_fcos_reader.yml', 7 | ] 8 | 9 | weights: output/fcos_r50_fpn_arsl_360k_coco_full/model_final 10 | 11 | #semi detector type 12 | ssod_method: ARSL 13 | -------------------------------------------------------------------------------- /configs/semi_det/arsl/arsl_fcos_r50_fpn_coco_semi001.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../_base_/coco_detection_percent_1.yml', 3 | '../../runtime.yml', 4 | '_base_/arsl_fcos_r50_fpn.yml', 5 | '_base_/optimizer_90k.yml', 6 | '_base_/arsl_fcos_reader.yml', 7 | ] 8 | 9 | weights: output/arsl_fcos_r50_fpn_coco_semi001/model_final 10 | 11 | #semi detector type 12 | ssod_method: ARSL 13 | -------------------------------------------------------------------------------- /configs/semi_det/arsl/arsl_fcos_r50_fpn_coco_semi005.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../_base_/coco_detection_percent_5.yml', 3 | '../../runtime.yml', 4 | '_base_/arsl_fcos_r50_fpn.yml', 5 | '_base_/optimizer_90k.yml', 6 | '_base_/arsl_fcos_reader.yml', 7 | ] 8 | 9 | weights: output/arsl_fcos_r50_fpn_coco_semi005/model_final 10 | 11 | #semi detector type 12 | ssod_method: ARSL 13 | -------------------------------------------------------------------------------- /configs/semi_det/arsl/arsl_fcos_r50_fpn_coco_semi010.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../_base_/coco_detection_percent_10.yml', 3 | '../../runtime.yml', 4 | '_base_/arsl_fcos_r50_fpn.yml', 5 | '_base_/optimizer_360k.yml', 6 | '_base_/arsl_fcos_reader.yml', 7 | ] 8 | 9 | weights: output/arsl_fcos_r50_fpn_coco_semi010/model_final 10 | 11 | #semi detector type 12 | ssod_method: ARSL 13 | -------------------------------------------------------------------------------- /configs/semi_det/baseline/fcos_r50_fpn_2x_coco_sup005.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../fcos/fcos_r50_fpn_iou_multiscale_2x_coco.yml', 3 | ] 4 | log_iter: 50 5 | snapshot_epoch: 2 6 | weights: output/fcos_r50_fpn_2x_coco_sup005/model_final 7 | 8 | 9 | TrainDataset: 10 | !COCODataSet 11 | image_dir: train2017 12 | anno_path: semi_annotations/instances_train2017.1@5.json 13 | dataset_dir: dataset/coco 14 | data_fields: ['image', 'gt_bbox', 'gt_class'] 15 | 16 | 17 | epoch: 24 18 | LearningRate: 19 | base_lr: 0.01 20 | schedulers: 21 | - !PiecewiseDecay 22 | gamma: 0.1 23 | milestones: [16, 22] 24 | - !LinearWarmup 25 | start_factor: 0.001 26 | epochs: 1 27 | -------------------------------------------------------------------------------- /configs/semi_det/baseline/fcos_r50_fpn_2x_coco_sup010.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../fcos/fcos_r50_fpn_iou_multiscale_2x_coco.yml', 3 | ] 4 | log_iter: 50 5 | snapshot_epoch: 2 6 | weights: output/fcos_r50_fpn_2x_coco_sup010/model_final 7 | 8 | 9 | TrainDataset: 10 | !COCODataSet 11 | image_dir: train2017 12 | anno_path: semi_annotations/instances_train2017.1@10.json 13 | dataset_dir: dataset/coco 14 | data_fields: ['image', 'gt_bbox', 'gt_class'] 15 | 16 | 17 | epoch: 24 18 | LearningRate: 19 | base_lr: 0.01 20 | schedulers: 21 | - !PiecewiseDecay 22 | gamma: 0.1 23 | milestones: [16, 22] 24 | - !LinearWarmup 25 | start_factor: 0.001 26 | epochs: 1 27 | -------------------------------------------------------------------------------- /configs/semi_det/baseline/retinanet_r50_fpn_2x_coco_sup005.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../retinanet/retinanet_r50_fpn_2x_coco.yml', 3 | ] 4 | log_iter: 50 5 | snapshot_epoch: 2 6 | weights: output/retinanet_r50_fpn_2x_coco_sup005/model_final 7 | 8 | 9 | TrainDataset: 10 | !COCODataSet 11 | image_dir: train2017 12 | anno_path: semi_annotations/instances_train2017.1@5.json 13 | dataset_dir: dataset/coco 14 | data_fields: ['image', 'gt_bbox', 'gt_class'] 15 | 16 | 17 | epoch: 24 18 | LearningRate: 19 | base_lr: 0.01 20 | schedulers: 21 | - !PiecewiseDecay 22 | gamma: 0.1 23 | milestones: [16, 22] 24 | - !LinearWarmup 25 | start_factor: 0.001 26 | epochs: 1 27 | -------------------------------------------------------------------------------- /configs/semi_det/baseline/retinanet_r50_fpn_2x_coco_sup010.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../retinanet/retinanet_r50_fpn_2x_coco.yml', 3 | ] 4 | log_iter: 50 5 | snapshot_epoch: 2 6 | weights: output/retinanet_r50_fpn_2x_coco_sup010/model_final 7 | 8 | 9 | TrainDataset: 10 | !COCODataSet 11 | image_dir: train2017 12 | anno_path: semi_annotations/instances_train2017.1@10.json 13 | dataset_dir: dataset/coco 14 | data_fields: ['image', 'gt_bbox', 'gt_class'] 15 | 16 | 17 | epoch: 24 18 | LearningRate: 19 | base_lr: 0.01 20 | schedulers: 21 | - !PiecewiseDecay 22 | gamma: 0.1 23 | milestones: [16, 22] 24 | - !LinearWarmup 25 | start_factor: 0.001 26 | epochs: 1 27 | -------------------------------------------------------------------------------- /configs/slim/distill/gfl_ld_distill.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../gfl/gfl_r18vd_1x_coco.yml', 3 | ] 4 | 5 | # teacher pretrain model 6 | pretrain_weights: https://paddledet.bj.bcebos.com/models/gfl_r101vd_fpn_mstrain_2x_coco.pdparams 7 | 8 | slim: Distill 9 | slim_method: LD 10 | 11 | ResNet: 12 | depth: 101 13 | variant: d 14 | norm_type: bn 15 | freeze_at: 0 16 | return_idx: [1,2,3] 17 | num_stages: 4 18 | 19 | TrainReader: 20 | batch_transforms: 21 | - PadBatch: {pad_to_stride: 32} 22 | - Gt2GFLTarget: 23 | downsample_ratios: [8, 16, 32, 64, 128] 24 | grid_cell_scale: 8 25 | compute_vlr_region: True -------------------------------------------------------------------------------- /configs/slim/distill/gfl_r101vd_fpn_coco_distill_cwd.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../gfl/gfl_r101vd_fpn_mstrain_2x_coco.yml', 3 | ] 4 | 5 | pretrain_weights: https://paddledet.bj.bcebos.com/models/gfl_r101vd_fpn_mstrain_2x_coco.pdparams 6 | 7 | slim: Distill 8 | slim_method: CWD 9 | distill_loss: CWDFeatureLoss 10 | distill_loss_name: ['cls_f_4', 'cls_f_3', 'cls_f_2', 'cls_f_1', 'cls_f_0'] 11 | 12 | CWDFeatureLoss: 13 | student_channels: 80 14 | teacher_channels: 80 15 | tau: 1.0 16 | weight: 5.0 17 | -------------------------------------------------------------------------------- /configs/slim/distill/retinanet_resnet101_coco_distill.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../retinanet/retinanet_r101_fpn_2x_coco.yml', 3 | ] 4 | 5 | pretrain_weights: https://paddledet.bj.bcebos.com/models/retinanet_r101_fpn_2x_coco.pdparams 6 | 7 | slim: Distill 8 | slim_method: FGD 9 | distill_loss: FGDFeatureLoss 10 | distill_loss_name: ['neck_f_4', 'neck_f_3', 'neck_f_2', 'neck_f_1', 'neck_f_0'] 11 | 12 | FGDFeatureLoss: 13 | student_channels: 256 14 | teacher_channels: 256 15 | temp: 0.5 16 | alpha_fgd: 0.001 17 | beta_fgd: 0.0005 18 | gamma_fgd: 0.0005 19 | lambda_fgd: 0.000005 20 | -------------------------------------------------------------------------------- /configs/slim/distill/retinanet_resnet101_coco_distill_cwd.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../retinanet/retinanet_r101_fpn_2x_coco.yml', 3 | ] 4 | 5 | pretrain_weights: https://paddledet.bj.bcebos.com/models/retinanet_r101_fpn_2x_coco.pdparams 6 | 7 | 8 | slim: Distill 9 | slim_method: CWD 10 | distill_loss: CWDFeatureLoss 11 | distill_loss_name: ['cls_f_4', 'cls_f_3', 'cls_f_2', 'cls_f_1', 'cls_f_0'] 12 | 13 | CWDFeatureLoss: 14 | student_channels: 80 15 | teacher_channels: 80 16 | tau: 1.0 17 | weight: 5.0 18 | -------------------------------------------------------------------------------- /configs/slim/distill/yolov3_mobilenet_v1_coco_distill.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../yolov3/yolov3_r34_270e_coco.yml', 3 | ] 4 | 5 | pretrain_weights: https://paddledet.bj.bcebos.com/models/yolov3_r34_270e_coco.pdparams 6 | 7 | 8 | slim: Distill 9 | distill_loss: DistillYOLOv3Loss 10 | 11 | DistillYOLOv3Loss: 12 | weight: 1000 13 | -------------------------------------------------------------------------------- /configs/slim/post_quant/mask_rcnn_r50_fpn_1x_coco_ptq.yml: -------------------------------------------------------------------------------- 1 | weights: https://paddledet.bj.bcebos.com/models/mask_rcnn_r50_fpn_1x_coco.pdparams 2 | slim: PTQ 3 | 4 | PTQ: 5 | ptq_config: { 6 | 'activation_quantizer': 'HistQuantizer', 7 | 'upsample_bins': 127, 8 | 'hist_percent': 0.999} 9 | quant_batch_num: 10 10 | fuse: True 11 | -------------------------------------------------------------------------------- /configs/slim/post_quant/mcfairmot_ptq.yml: -------------------------------------------------------------------------------- 1 | weights: https://paddledet.bj.bcebos.com/models/mot/mcfairmot_dla34_30e_1088x608_visdrone_vehicle_bytetracker.pdparams 2 | slim: PTQ 3 | 4 | PTQ: 5 | ptq_config: { 6 | 'activation_quantizer': 'HistQuantizer', 7 | 'upsample_bins': 127, 8 | 'hist_percent': 0.999} 9 | quant_batch_num: 10 10 | fuse: True 11 | -------------------------------------------------------------------------------- /configs/slim/post_quant/picodet_s_ptq.yml: -------------------------------------------------------------------------------- 1 | weights: https://paddledet.bj.bcebos.com/models/picodet_s_320_coco.pdparams 2 | slim: PTQ 3 | 4 | PTQ: 5 | ptq_config: { 6 | 'activation_quantizer': 'HistQuantizer', 7 | 'upsample_bins': 127, 8 | 'hist_percent': 0.999} 9 | quant_batch_num: 10 10 | fuse: True 11 | -------------------------------------------------------------------------------- /configs/slim/post_quant/ppyolo_mbv3_large_ptq.yml: -------------------------------------------------------------------------------- 1 | weights: https://paddledet.bj.bcebos.com/models/ppyolo_mbv3_large_coco.pdparams 2 | slim: PTQ 3 | 4 | PTQ: 5 | ptq_config: { 6 | 'activation_quantizer': 'HistQuantizer', 7 | 'upsample_bins': 127, 8 | 'hist_percent': 0.999} 9 | quant_batch_num: 10 10 | fuse: True 11 | -------------------------------------------------------------------------------- /configs/slim/post_quant/ppyolo_r50vd_dcn_ptq.yml: -------------------------------------------------------------------------------- 1 | weights: https://paddledet.bj.bcebos.com/models/ppyolo_r50vd_dcn_1x_coco.pdparams 2 | slim: PTQ 3 | 4 | PTQ: 5 | ptq_config: { 6 | 'activation_quantizer': 'HistQuantizer', 7 | 'upsample_bins': 127, 8 | 'hist_percent': 0.999} 9 | quant_batch_num: 10 10 | fuse: True 11 | -------------------------------------------------------------------------------- /configs/slim/post_quant/ppyoloe_crn_s_300e_coco_ptq.yml: -------------------------------------------------------------------------------- 1 | weights: https://paddledet.bj.bcebos.com/models/ppyoloe_crn_s_300e_coco.pdparams 2 | slim: PTQ 3 | 4 | PTQ: 5 | ptq_config: { 6 | 'activation_quantizer': 'HistQuantizer', 7 | 'upsample_bins': 127, 8 | 'hist_percent': 0.999} 9 | quant_batch_num: 10 10 | fuse: True 11 | -------------------------------------------------------------------------------- /configs/slim/post_quant/tinypose_128x96_ptq.yml: -------------------------------------------------------------------------------- 1 | weights: https://paddledet.bj.bcebos.com/models/keypoint/tinypose_128x96.pdparams 2 | slim: PTQ 3 | 4 | PTQ: 5 | ptq_config: { 6 | 'activation_quantizer': 'HistQuantizer', 7 | 'upsample_bins': 127, 8 | 'hist_percent': 0.999} 9 | quant_batch_num: 10 10 | fuse: True 11 | -------------------------------------------------------------------------------- /configs/slim/post_quant/yolov3_darknet53_ptq.yml: -------------------------------------------------------------------------------- 1 | weights: https://paddledet.bj.bcebos.com/models/yolov3_darknet53_270e_coco.pdparams 2 | slim: PTQ 3 | 4 | PTQ: 5 | ptq_config: { 6 | 'activation_quantizer': 'HistQuantizer', 7 | 'upsample_bins': 127, 8 | 'hist_percent': 0.999} 9 | quant_batch_num: 10 10 | fuse: True 11 | -------------------------------------------------------------------------------- /configs/slim/prune/picodet_m_unstructured_prune_75.yml: -------------------------------------------------------------------------------- 1 | pretrain_weights: https://paddledet.bj.bcebos.com/models/picodet_m_320_coco.pdparams 2 | slim: UnstructuredPruner 3 | 4 | UnstructuredPruner: 5 | stable_epochs: 0 6 | pruning_epochs: 150 7 | tunning_epochs: 150 8 | pruning_steps: 300 9 | ratio: 0.75 10 | initial_ratio: 0.15 11 | prune_params_type: conv1x1_only 12 | -------------------------------------------------------------------------------- /configs/slim/prune/picodet_m_unstructured_prune_85.yml: -------------------------------------------------------------------------------- 1 | pretrain_weights: https://paddledet.bj.bcebos.com/models/picodet_m_320_coco.pdparams 2 | slim: UnstructuredPruner 3 | 4 | UnstructuredPruner: 5 | stable_epochs: 0 6 | pruning_epochs: 150 7 | tunning_epochs: 150 8 | pruning_steps: 300 9 | ratio: 0.85 10 | initial_ratio: 0.20 11 | prune_params_type: conv1x1_only 12 | -------------------------------------------------------------------------------- /configs/slim/prune/ppyolo_mbv3_large_prune_fpgm.yml: -------------------------------------------------------------------------------- 1 | pretrain_weights: https://paddledet.bj.bcebos.com/models/ppyolo_mbv3_large_coco.pdparams 2 | slim: Pruner 3 | 4 | Pruner: 5 | criterion: fpgm 6 | pruned_params: ['conv2d_62.w_0', 'conv2d_63.w_0', 'conv2d_64.w_0', 7 | 'conv2d_65.w_0', 'conv2d_66.w_0', 'conv2d_67.w_0'] 8 | pruned_ratios: [0.75, 0.75, 0.75, 0.75, 0.75, 0.75] 9 | print_params: True 10 | -------------------------------------------------------------------------------- /configs/slim/quant/mask_rcnn_r50_fpn_1x_qat.yml: -------------------------------------------------------------------------------- 1 | pretrain_weights: https://paddledet.bj.bcebos.com/models/mask_rcnn_r50_fpn_1x_coco.pdparams 2 | slim: QAT 3 | 4 | QAT: 5 | quant_config: { 6 | 'weight_quantize_type': 'channel_wise_abs_max', 'activation_quantize_type': 'moving_average_abs_max', 7 | 'weight_bits': 8, 'activation_bits': 8, 'dtype': 'int8', 'window_size': 10000, 'moving_rate': 0.9, 8 | 'quantizable_layer_type': ['Conv2D', 'Linear']} 9 | print_model: True 10 | 11 | 12 | epoch: 5 13 | 14 | LearningRate: 15 | base_lr: 0.001 16 | schedulers: 17 | - !PiecewiseDecay 18 | gamma: 0.1 19 | milestones: [3, 4] 20 | - !LinearWarmup 21 | start_factor: 0.001 22 | steps: 100 23 | -------------------------------------------------------------------------------- /configs/slim/quant/ppyolo_mbv3_large_qat.yml: -------------------------------------------------------------------------------- 1 | pretrain_weights: https://paddledet.bj.bcebos.com/models/ppyolo_mbv3_large_coco.pdparams 2 | slim: QAT 3 | 4 | QAT: 5 | quant_config: { 6 | 'weight_quantize_type': 'channel_wise_abs_max', 'activation_quantize_type': 'moving_average_abs_max', 7 | 'weight_bits': 8, 'activation_bits': 8, 'dtype': 'int8', 'window_size': 10000, 'moving_rate': 0.99, 8 | 'quantizable_layer_type': ['Conv2D', 'Linear']} 9 | print_model: True 10 | 11 | PPYOLOFPN: 12 | in_channels: [160, 368] 13 | coord_conv: true 14 | conv_block_num: 0 15 | spp: true 16 | drop_block: false 17 | -------------------------------------------------------------------------------- /configs/slim/quant/ssd_mobilenet_v1_qat.yml: -------------------------------------------------------------------------------- 1 | pretrain_weights: https://paddlemodels.bj.bcebos.com/object_detection/dygraph/ssd_mobilenet_v1_300_120e_voc.pdparams 2 | slim: QAT 3 | 4 | QAT: 5 | quant_config: { 6 | 'weight_quantize_type': 'channel_wise_abs_max', 'activation_quantize_type': 'moving_average_abs_max', 7 | 'weight_bits': 8, 'activation_bits': 8, 'dtype': 'int8', 'window_size': 10000, 'moving_rate': 0.9, 8 | 'quantizable_layer_type': ['Conv2D', 'Linear']} 9 | print_model: True 10 | -------------------------------------------------------------------------------- /configs/slim/quant/yolov3_mobilenet_v1_qat.yml: -------------------------------------------------------------------------------- 1 | # Weights of yolov3_mobilenet_v1_coco 2 | pretrain_weights: https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v1_270e_coco.pdparams 3 | slim: QAT 4 | 5 | QAT: 6 | quant_config: { 7 | 'weight_quantize_type': 'channel_wise_abs_max', 'activation_quantize_type': 'moving_average_abs_max', 8 | 'weight_bits': 8, 'activation_bits': 8, 'dtype': 'int8', 'window_size': 10000, 'moving_rate': 0.9, 9 | 'quantizable_layer_type': ['Conv2D', 'Linear']} 10 | print_model: True 11 | -------------------------------------------------------------------------------- /configs/smalldet/_base_/DOTA_sliced_500_025_detection.yml: -------------------------------------------------------------------------------- 1 | metric: COCO 2 | num_classes: 15 3 | 4 | TrainDataset: 5 | !COCODataSet 6 | image_dir: train_images_500_025 7 | anno_path: train_500_025.json 8 | dataset_dir: dataset/dota_sliced 9 | data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd'] 10 | 11 | EvalDataset: 12 | !COCODataSet 13 | image_dir: val_images_500_025 14 | anno_path: val_500_025.json 15 | dataset_dir: dataset/dota_sliced 16 | 17 | TestDataset: 18 | !ImageFolder 19 | anno_path: val_500_025.json 20 | dataset_dir: dataset/dota_sliced 21 | -------------------------------------------------------------------------------- /configs/smalldet/_base_/visdrone_sliced_640_025_detection.yml: -------------------------------------------------------------------------------- 1 | metric: COCO 2 | num_classes: 10 3 | 4 | TrainDataset: 5 | !COCODataSet 6 | image_dir: train_images_640_025 7 | anno_path: train_640_025.json 8 | dataset_dir: dataset/visdrone_sliced 9 | data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd'] 10 | 11 | EvalDataset: 12 | !COCODataSet 13 | image_dir: val_images_640_025 14 | anno_path: val_640_025.json 15 | dataset_dir: dataset/visdrone_sliced 16 | 17 | TestDataset: 18 | !ImageFolder 19 | anno_path: val_640_025.json 20 | dataset_dir: dataset/visdrone_sliced 21 | -------------------------------------------------------------------------------- /configs/smalldet/_base_/xview_sliced_400_025_detection.yml: -------------------------------------------------------------------------------- 1 | metric: COCO 2 | num_classes: 60 3 | 4 | TrainDataset: 5 | !COCODataSet 6 | image_dir: train_images_400_025 7 | anno_path: train_400_025.json 8 | dataset_dir: dataset/xview_sliced 9 | data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd'] 10 | 11 | EvalDataset: 12 | !COCODataSet 13 | image_dir: val_images_400_025 14 | anno_path: val_400_025.json 15 | dataset_dir: dataset/xview_sliced 16 | 17 | TestDataset: 18 | !ImageFolder 19 | anno_path: val_400_025.json 20 | dataset_dir: dataset/xview_sliced 21 | -------------------------------------------------------------------------------- /configs/smalldet/ppyoloe_crn_l_80e_sliced_visdrone_640_025_slice_infer.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'ppyoloe_crn_l_80e_sliced_visdrone_640_025.yml', 3 | ] 4 | weights: https://paddledet.bj.bcebos.com/models/ppyoloe_crn_l_80e_sliced_visdrone_640_025.pdparams 5 | 6 | 7 | # very slow, better to use the determined weight (xx. pdparams) to evaluate separately, rather than evaling during training 8 | # if you want to eval during training, change SlicedCOCODataSet to COCODataSet, then delete sliced_size and overlap_ratio 9 | EvalDataset: 10 | !SlicedCOCODataSet 11 | image_dir: VisDrone2019-DET-val 12 | anno_path: val.json 13 | dataset_dir: dataset/visdrone 14 | sliced_size: [640, 640] 15 | overlap_ratio: [0.25, 0.25] 16 | -------------------------------------------------------------------------------- /configs/smalldet/visdrone/ppyoloe_crn_s_p2_alpha_80e_visdrone.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'ppyoloe_crn_s_80e_visdrone.yml', 3 | ] 4 | log_iter: 100 5 | snapshot_epoch: 10 6 | weights: output/ppyoloe_crn_s_p2_alpha_80e_visdrone/model_final 7 | pretrain_weights: https://paddledet.bj.bcebos.com/models/ppyoloe_crn_s_300e_coco.pdparams 8 | 9 | 10 | TrainReader: 11 | batch_size: 4 12 | 13 | EvalReader: 14 | batch_size: 1 15 | 16 | TestReader: 17 | batch_size: 1 18 | fuse_normalize: True 19 | 20 | 21 | LearningRate: 22 | base_lr: 0.005 23 | 24 | CSPResNet: 25 | return_idx: [0, 1, 2, 3] 26 | use_alpha: True 27 | 28 | CustomCSPPAN: 29 | out_channels: [768, 384, 192, 64] 30 | 31 | PPYOLOEHead: 32 | fpn_strides: [32, 16, 8, 4] 33 | -------------------------------------------------------------------------------- /configs/smrt/images/00362.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/smrt/images/00362.jpg -------------------------------------------------------------------------------- /configs/smrt/images/Board_diaojiao_1591.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/smrt/images/Board_diaojiao_1591.png -------------------------------------------------------------------------------- /configs/smrt/images/UpCoa_liewen_163.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/smrt/images/UpCoa_liewen_163.png -------------------------------------------------------------------------------- /configs/smrt/images/lvjian1_0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/smrt/images/lvjian1_0.jpg -------------------------------------------------------------------------------- /configs/smrt/images/lvjian1_10.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/smrt/images/lvjian1_10.jpg -------------------------------------------------------------------------------- /configs/smrt/images/renche_00002.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/smrt/images/renche_00002.jpg -------------------------------------------------------------------------------- /configs/smrt/images/renche_00204.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/smrt/images/renche_00204.jpg -------------------------------------------------------------------------------- /configs/sniper/faster_rcnn_r50_fpn_1x_sniper_visdrone.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/sniper_visdrone_detection.yml', 3 | '../runtime.yml', 4 | '../faster_rcnn/_base_/faster_rcnn_r50_fpn.yml', 5 | '../faster_rcnn/_base_/optimizer_1x.yml', 6 | '_base_/faster_fpn_reader.yml', 7 | ] 8 | weights: output/faster_rcnn_r50_fpn_1x_sniper_visdrone/model_final 9 | find_unused_parameters: true 10 | -------------------------------------------------------------------------------- /configs/solov2/_base_/optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [8, 11] 9 | - !LinearWarmup 10 | start_factor: 0. 11 | steps: 1000 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0001 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/solov2/solov2_r50_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_instance.yml', 3 | '../runtime.yml', 4 | '_base_/solov2_r50_fpn.yml', 5 | '_base_/optimizer_1x.yml', 6 | '_base_/solov2_reader.yml', 7 | ] 8 | weights: output/solov2_r50_fpn_1x_coco/model_final 9 | -------------------------------------------------------------------------------- /configs/sparse_rcnn/_base_/optimizer_3x.yml: -------------------------------------------------------------------------------- 1 | epoch: 36 2 | 3 | LearningRate: 4 | base_lr: 0.000025 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [28, 34] 9 | - !LinearWarmup 10 | start_factor: 0.01 11 | steps: 1000 12 | 13 | OptimizerBuilder: 14 | clip_grad_by_norm: 1.0 15 | optimizer: 16 | type: AdamW 17 | weight_decay: 0.0001 18 | -------------------------------------------------------------------------------- /configs/sparse_rcnn/sparse_rcnn_r50_fpn_3x_pro100_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/sparse_rcnn_r50_fpn.yml', 5 | '_base_/optimizer_3x.yml', 6 | '_base_/sparse_rcnn_reader.yml', 7 | ] 8 | 9 | num_classes: 80 10 | weights: output/sparse_rcnn_r50_fpn_3x_pro100_coco/model_final 11 | -------------------------------------------------------------------------------- /configs/sparse_rcnn/sparse_rcnn_r50_fpn_3x_pro300_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/sparse_rcnn_r50_fpn.yml', 5 | '_base_/optimizer_3x.yml', 6 | '_base_/sparse_rcnn_reader.yml', 7 | ] 8 | 9 | num_classes: 80 10 | weights: output/sparse_rcnn_r50_fpn_3x_pro300_coco/model_final 11 | 12 | snapshot_epoch: 1 13 | 14 | 15 | SparseRCNNHead: 16 | num_proposals: 300 17 | 18 | SparsePostProcess: 19 | num_proposals: 300 20 | -------------------------------------------------------------------------------- /configs/ssd/_base_/optimizer_120e.yml: -------------------------------------------------------------------------------- 1 | epoch: 120 2 | 3 | LearningRate: 4 | base_lr: 0.001 5 | schedulers: 6 | - !PiecewiseDecay 7 | milestones: [40, 60, 80, 100] 8 | gamma: [0.5, 0.5, 0.4, 0.1] 9 | use_warmup: false 10 | 11 | OptimizerBuilder: 12 | optimizer: 13 | momentum: 0.0 14 | type: RMSProp 15 | regularizer: 16 | factor: 0.00005 17 | type: L2 18 | -------------------------------------------------------------------------------- /configs/ssd/_base_/optimizer_1700e.yml: -------------------------------------------------------------------------------- 1 | epoch: 1700 2 | 3 | LearningRate: 4 | base_lr: 0.4 5 | schedulers: 6 | - !CosineDecay 7 | max_epochs: 1700 8 | - !LinearWarmup 9 | start_factor: 0.3333333333333333 10 | steps: 2000 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | momentum: 0.9 15 | type: Momentum 16 | regularizer: 17 | factor: 0.0005 18 | type: L2 19 | -------------------------------------------------------------------------------- /configs/ssd/_base_/optimizer_240e.yml: -------------------------------------------------------------------------------- 1 | epoch: 240 2 | 3 | LearningRate: 4 | base_lr: 0.001 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: 9 | - 160 10 | - 200 11 | - !LinearWarmup 12 | start_factor: 0.3333333333333333 13 | steps: 500 14 | 15 | OptimizerBuilder: 16 | optimizer: 17 | momentum: 0.9 18 | type: Momentum 19 | regularizer: 20 | factor: 0.0005 21 | type: L2 22 | -------------------------------------------------------------------------------- /configs/ssd/_base_/optimizer_70e.yml: -------------------------------------------------------------------------------- 1 | epoch: 70 2 | 3 | LearningRate: 4 | base_lr: 0.05 5 | schedulers: 6 | - !PiecewiseDecay 7 | milestones: [48, 60] 8 | gamma: [0.1, 0.1] 9 | use_warmup: false 10 | 11 | OptimizerBuilder: 12 | optimizer: 13 | momentum: 0.9 14 | type: Momentum 15 | regularizer: 16 | factor: 0.0005 17 | type: L2 18 | -------------------------------------------------------------------------------- /configs/ssd/ssd_mobilenet_v1_300_120e_voc.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/voc.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_120e.yml', 5 | '_base_/ssd_mobilenet_v1_300.yml', 6 | '_base_/ssd_mobilenet_reader.yml', 7 | ] 8 | weights: output/ssd_mobilenet_v1_300_120e_voc/model_final 9 | 10 | # set collate_batch to false because ground-truth info is needed 11 | # on voc dataset and should not collate data in batch when batch size 12 | # is larger than 1. 13 | EvalReader: 14 | collate_batch: false 15 | -------------------------------------------------------------------------------- /configs/ssd/ssd_r34_70e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_70e.yml', 5 | '_base_/ssd_r34_300.yml', 6 | '_base_/ssd_r34_reader.yml', 7 | ] 8 | weights: output/ssd_r34_70e_coco/model_final 9 | 10 | log_iter: 100 11 | snapshot_epoch: 5 12 | -------------------------------------------------------------------------------- /configs/ssd/ssd_vgg16_300_240e_voc.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/voc.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_240e.yml', 5 | '_base_/ssd_vgg16_300.yml', 6 | '_base_/ssd_reader.yml', 7 | ] 8 | weights: output/ssd_vgg16_300_240e_voc/model_final 9 | 10 | # set collate_batch to false because ground-truth info is needed 11 | # on voc dataset and should not collate data in batch when batch size 12 | # is larger than 1. 13 | EvalReader: 14 | collate_batch: false 15 | -------------------------------------------------------------------------------- /configs/ssd/ssdlite_ghostnet_320_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1700e.yml', 5 | '_base_/ssdlite_ghostnet_320.yml', 6 | '_base_/ssdlite320_reader.yml', 7 | ] 8 | weights: output/ssdlite_ghostnet_320_coco/model_final 9 | 10 | epoch: 1700 11 | 12 | LearningRate: 13 | base_lr: 0.2 14 | schedulers: 15 | - !CosineDecay 16 | max_epochs: 1700 17 | - !LinearWarmup 18 | start_factor: 0.33333 19 | steps: 2000 20 | 21 | OptimizerBuilder: 22 | optimizer: 23 | momentum: 0.9 24 | type: Momentum 25 | regularizer: 26 | factor: 0.0005 27 | type: L2 28 | -------------------------------------------------------------------------------- /configs/ssd/ssdlite_mobilenet_v1_300_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1700e.yml', 5 | '_base_/ssdlite_mobilenet_v1_300.yml', 6 | '_base_/ssdlite300_reader.yml', 7 | ] 8 | weights: output/ssdlite_mobilenet_v1_300_coco/model_final 9 | -------------------------------------------------------------------------------- /configs/ssd/ssdlite_mobilenet_v3_large_320_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1700e.yml', 5 | '_base_/ssdlite_mobilenet_v3_large_320.yml', 6 | '_base_/ssdlite320_reader.yml', 7 | ] 8 | weights: output/ssdlite_mobilenet_v3_large_320_coco/model_final 9 | -------------------------------------------------------------------------------- /configs/ssd/ssdlite_mobilenet_v3_small_320_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1700e.yml', 5 | '_base_/ssdlite_mobilenet_v3_small_320.yml', 6 | '_base_/ssdlite320_reader.yml', 7 | ] 8 | weights: output/ssdlite_mobilenet_v3_small_320_coco/model_final 9 | -------------------------------------------------------------------------------- /configs/tood/_base_/optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [8, 11] 9 | - !LinearWarmup 10 | start_factor: 0.001 11 | steps: 500 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0001 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/tood/tood_r50_fpn_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/tood_r50_fpn.yml', 5 | '_base_/optimizer_1x.yml', 6 | '_base_/tood_reader.yml', 7 | ] 8 | 9 | weights: output/tood_r50_fpn_1x_coco/model_final 10 | find_unused_parameters: True 11 | log_iter: 100 12 | -------------------------------------------------------------------------------- /configs/ttfnet/_base_/optimizer_10x.yml: -------------------------------------------------------------------------------- 1 | epoch: 120 2 | 3 | LearningRate: 4 | base_lr: 0.015 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [80, 110] 9 | - !LinearWarmup 10 | start_factor: 0.2 11 | steps: 500 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0004 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/ttfnet/_base_/optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.015 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [8, 11] 9 | - !LinearWarmup 10 | start_factor: 0.2 11 | steps: 500 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0004 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/ttfnet/_base_/optimizer_20x.yml: -------------------------------------------------------------------------------- 1 | epoch: 240 2 | 3 | LearningRate: 4 | base_lr: 0.015 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [160, 220] 9 | - !LinearWarmup 10 | start_factor: 0.2 11 | steps: 1000 12 | 13 | OptimizerBuilder: 14 | clip_grad_by_norm: 35 15 | optimizer: 16 | momentum: 0.9 17 | type: Momentum 18 | regularizer: 19 | factor: 0.0004 20 | type: L2 21 | -------------------------------------------------------------------------------- /configs/ttfnet/pafnet_10x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_10x.yml', 5 | '_base_/pafnet.yml', 6 | '_base_/pafnet_reader.yml', 7 | ] 8 | weights: output/pafnet_10x_coco/model_final 9 | -------------------------------------------------------------------------------- /configs/ttfnet/pafnet_lite_mobilenet_v3_20x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_20x.yml', 5 | '_base_/pafnet_lite.yml', 6 | '_base_/pafnet_lite_reader.yml', 7 | ] 8 | weights: output/pafnet_lite_mobilenet_v3_10x_coco/model_final 9 | -------------------------------------------------------------------------------- /configs/ttfnet/ttfnet_darknet53_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_1x.yml', 5 | '_base_/ttfnet_darknet53.yml', 6 | '_base_/ttfnet_reader.yml', 7 | ] 8 | weights: output/ttfnet_darknet53_1x_coco/model_final 9 | -------------------------------------------------------------------------------- /configs/vitdet/_base_/optimizer_base_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.0001 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [9, 11] 9 | - !LinearWarmup 10 | start_factor: 0.001 11 | steps: 1000 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | type: AdamWDL 16 | betas: [0.9, 0.999] 17 | layer_decay: 0.75 18 | weight_decay: 0.02 19 | num_layers: 12 20 | filter_bias_and_bn: True 21 | skip_decay_names: ['pos_embed', 'cls_token'] 22 | set_param_lr_func: 'layerwise_lr_decay' 23 | -------------------------------------------------------------------------------- /configs/vitdet/_base_/optimizer_base_30e.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/configs/vitdet/_base_/optimizer_base_30e.yml -------------------------------------------------------------------------------- /configs/vitdet/_base_/optimizer_base_36e.yml: -------------------------------------------------------------------------------- 1 | 2 | epoch: 36 3 | 4 | LearningRate: 5 | base_lr: 0.0001 6 | schedulers: 7 | - !CosineDecay 8 | max_epochs: 36 9 | min_lr_ratio: 0.1 # 0.1 10 | - !LinearWarmup 11 | start_factor: 0.001 12 | epochs: 1 13 | 14 | 15 | OptimizerBuilder: 16 | clip_grad_by_norm: 0.1 17 | regularizer: false 18 | optimizer: 19 | type: AdamW 20 | weight_decay: 0.0001 21 | -------------------------------------------------------------------------------- /configs/yolof/_base_/optimizer_1x.yml: -------------------------------------------------------------------------------- 1 | epoch: 12 2 | 3 | LearningRate: 4 | base_lr: 0.06 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: [8, 11] 9 | - !LinearWarmup 10 | start_factor: 0.00066 11 | steps: 1500 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | momentum: 0.9 16 | type: Momentum 17 | regularizer: 18 | factor: 0.0001 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolof/yolof_r50_c5_1x_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_1x.yml', 5 | './_base_/yolof_r50_c5.yml', 6 | './_base_/yolof_reader.yml' 7 | ] 8 | log_iter: 50 9 | snapshot_epoch: 1 10 | weights: output/yolof_r50_c5_1x_coco/model_final 11 | -------------------------------------------------------------------------------- /configs/yolov3/_base_/optimizer_270e.yml: -------------------------------------------------------------------------------- 1 | epoch: 270 2 | 3 | LearningRate: 4 | base_lr: 0.001 5 | schedulers: 6 | - !PiecewiseDecay 7 | gamma: 0.1 8 | milestones: 9 | - 216 10 | - 243 11 | - !LinearWarmup 12 | start_factor: 0. 13 | steps: 4000 14 | 15 | OptimizerBuilder: 16 | optimizer: 17 | momentum: 0.9 18 | type: Momentum 19 | regularizer: 20 | factor: 0.0005 21 | type: L2 22 | -------------------------------------------------------------------------------- /configs/yolov3/_base_/optimizer_40e.yml: -------------------------------------------------------------------------------- 1 | epoch: 40 2 | 3 | LearningRate: 4 | base_lr: 0.0001 5 | schedulers: 6 | - name: PiecewiseDecay 7 | gamma: 0.1 8 | milestones: 9 | - 32 10 | - 36 11 | - name: LinearWarmup 12 | start_factor: 0.3333333333333333 13 | steps: 100 14 | 15 | OptimizerBuilder: 16 | optimizer: 17 | momentum: 0.9 18 | type: Momentum 19 | regularizer: 20 | factor: 0.0005 21 | type: L2 22 | -------------------------------------------------------------------------------- /configs/yolov3/yolov3_darknet53_270e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_270e.yml', 5 | '_base_/yolov3_darknet53.yml', 6 | '_base_/yolov3_reader.yml', 7 | ] 8 | 9 | snapshot_epoch: 5 10 | weights: output/yolov3_darknet53_270e_coco/model_final 11 | -------------------------------------------------------------------------------- /configs/yolov3/yolov3_mobilenet_v1_270e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_270e.yml', 5 | '_base_/yolov3_mobilenet_v1.yml', 6 | '_base_/yolov3_reader.yml', 7 | ] 8 | 9 | snapshot_epoch: 5 10 | weights: output/yolov3_mobilenet_v1_270e_coco/model_final 11 | -------------------------------------------------------------------------------- /configs/yolov3/yolov3_mobilenet_v1_roadsign.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/roadsign_voc.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_40e.yml', 5 | '_base_/yolov3_mobilenet_v1.yml', 6 | '_base_/yolov3_reader.yml', 7 | ] 8 | pretrain_weights: https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v1_270e_coco.pdparams 9 | weights: output/yolov3_mobilenet_v1_roadsign/model_final 10 | 11 | YOLOv3Loss: 12 | ignore_thresh: 0.7 13 | label_smooth: true 14 | -------------------------------------------------------------------------------- /configs/yolov3/yolov3_mobilenet_v1_ssld_270e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_270e.yml', 5 | '_base_/yolov3_mobilenet_v1.yml', 6 | '_base_/yolov3_reader.yml', 7 | ] 8 | 9 | snapshot_epoch: 5 10 | pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/MobileNetV1_ssld_pretrained.pdparams 11 | weights: output/yolov3_mobilenet_v1_ssld_270e_coco/model_final 12 | -------------------------------------------------------------------------------- /configs/yolov3/yolov3_mobilenet_v3_large_270e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_270e.yml', 5 | '_base_/yolov3_mobilenet_v3_large.yml', 6 | '_base_/yolov3_reader.yml', 7 | ] 8 | 9 | snapshot_epoch: 5 10 | weights: output/yolov3_mobilenet_v3_large_270e_coco/model_final 11 | -------------------------------------------------------------------------------- /configs/yolov3/yolov3_r34_270e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_270e.yml', 5 | '_base_/yolov3_r34.yml', 6 | '_base_/yolov3_reader.yml', 7 | ] 8 | 9 | snapshot_epoch: 5 10 | weights: output/yolov3_r34_270e_coco/model_final 11 | -------------------------------------------------------------------------------- /configs/yolov3/yolov3_r50vd_dcn_270e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_270e.yml', 5 | '_base_/yolov3_r50vd_dcn.yml', 6 | '_base_/yolov3_reader.yml', 7 | ] 8 | 9 | snapshot_epoch: 5 10 | weights: output/yolov3_r50vd_dcn_270e_coco/model_final 11 | -------------------------------------------------------------------------------- /configs/yolov5/_base_/optimizer_300e.yml: -------------------------------------------------------------------------------- 1 | epoch: 300 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !YOLOv5LRDecay 7 | max_epochs: 300 8 | min_lr_ratio: 0.01 9 | - !ExpWarmup 10 | epochs: 3 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | type: Momentum 15 | momentum: 0.937 16 | use_nesterov: True 17 | regularizer: 18 | factor: 0.0005 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolov5/_base_/optimizer_300e_high.yml: -------------------------------------------------------------------------------- 1 | epoch: 300 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !YOLOv5LRDecay 7 | max_epochs: 300 8 | min_lr_ratio: 0.1 # 9 | - !ExpWarmup 10 | epochs: 3 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | type: Momentum 15 | momentum: 0.937 16 | use_nesterov: True 17 | regularizer: 18 | factor: 0.0005 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolov5/_base_/yolov5_cspresnet.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'yolov5_cspdarknet.yml', 3 | ] 4 | 5 | YOLOv5: 6 | backbone: CSPResNet 7 | neck: CustomCSPPAN 8 | yolo_head: YOLOv5Head 9 | post_process: ~ 10 | 11 | CSPResNet: 12 | layers: [3, 6, 6, 3] 13 | channels: [64, 128, 256, 512, 1024] 14 | return_idx: [1, 2, 3] 15 | use_large_stem: True 16 | 17 | CustomCSPPAN: 18 | out_channels: [768, 384, 192] 19 | stage_num: 1 20 | block_num: 3 21 | act: 'swish' 22 | spp: true 23 | -------------------------------------------------------------------------------- /configs/yolov5/_base_/yolov5p2_cspdarknet.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'yolov5_cspdarknet.yml', 3 | ] 4 | 5 | CSPDarkNet: 6 | arch: "P5" 7 | return_idx: [1, 2, 3, 4] 8 | 9 | YOLOv5Head: 10 | anchors: [[5, 6], [8, 14], [15, 11], 11 | [10, 13], [16, 30], [33, 23], 12 | [30, 61], [62, 45], [59, 119], 13 | [116, 90], [156, 198], [373, 326]] 14 | anchor_masks: [[0, 1, 2], [3, 4, 5], [6, 7, 8], [9, 10, 11]] 15 | stride: [4, 8, 16, 32] 16 | 17 | YOLOv5Loss: 18 | downsample_ratios: [4, 8, 16, 32] 19 | balance: [4.0, 1.0, 0.25, 0.06] 20 | -------------------------------------------------------------------------------- /configs/yolov5/_base_/yolov5p6_cspdarknet.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'yolov5_cspdarknet.yml', 3 | ] 4 | 5 | CSPDarkNet: 6 | arch: "P6" 7 | return_idx: [2, 3, 4, 5] 8 | 9 | YOLOv5Head: 10 | anchors: [[19, 27], [44, 40], [38, 94], 11 | [96, 68], [86, 152], [180, 137], 12 | [140, 301], [303, 264], [238, 542], 13 | [436, 615], [739, 380], [925, 792]] 14 | anchor_masks: [[0, 1, 2], [3, 4, 5], [6, 7, 8], [9, 10, 11]] 15 | stride: [8, 16, 32, 64] 16 | 17 | YOLOv5Loss: 18 | downsample_ratios: [8, 16, 32, 64] 19 | balance: [4.0, 1.0, 0.25, 0.06] 20 | box_weight: 0.05 21 | obj_weight: 0.7 22 | cls_weght: 0.3 23 | -------------------------------------------------------------------------------- /configs/yolov5/yolov5_l_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_high.yml', 5 | '_base_/yolov5_cspdarknet.yml', 6 | '_base_/yolov5_reader_high_aug.yml', 7 | ] 8 | depth_mult: 1.0 9 | width_mult: 1.0 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5_l_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | 19 | 20 | YOLOv5Loss: 21 | obj_weight: 0.7 22 | cls_weght: 0.3 23 | -------------------------------------------------------------------------------- /configs/yolov5/yolov5_m_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_high.yml', 5 | '_base_/yolov5_cspdarknet.yml', 6 | '_base_/yolov5_reader_high_aug.yml', 7 | ] 8 | depth_mult: 0.67 9 | width_mult: 0.75 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5_m_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | 19 | 20 | YOLOv5Loss: 21 | obj_weight: 0.7 22 | cls_weght: 0.3 23 | -------------------------------------------------------------------------------- /configs/yolov5/yolov5_n_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e.yml', 5 | '_base_/yolov5_cspdarknet.yml', 6 | '_base_/yolov5_reader.yml', 7 | ] 8 | depth_mult: 0.33 9 | width_mult: 0.25 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5_n_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | -------------------------------------------------------------------------------- /configs/yolov5/yolov5_s_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e.yml', 5 | '_base_/yolov5_cspdarknet.yml', 6 | '_base_/yolov5_reader.yml', 7 | ] 8 | depth_mult: 0.33 9 | width_mult: 0.50 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5_s_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | -------------------------------------------------------------------------------- /configs/yolov5/yolov5_x_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_high.yml', 5 | '_base_/yolov5_cspdarknet.yml', 6 | '_base_/yolov5_reader_high_aug.yml', 7 | ] 8 | depth_mult: 1.33 9 | width_mult: 1.25 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5_x_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | 19 | 20 | YOLOv5Loss: 21 | obj_weight: 0.7 22 | cls_weght: 0.3 23 | -------------------------------------------------------------------------------- /configs/yolov5/yolov5p6_l_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_high.yml', 5 | '_base_/yolov5p6_cspdarknet.yml', 6 | '_base_/yolov5p6_reader.yml', 7 | ] 8 | depth_mult: 1.0 9 | width_mult: 1.0 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5p6_l_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 8 # default 8 gpus, total bs = 64 18 | -------------------------------------------------------------------------------- /configs/yolov5/yolov5p6_m_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_high.yml', 5 | '_base_/yolov5p6_cspdarknet.yml', 6 | '_base_/yolov5p6_reader.yml', 7 | ] 8 | depth_mult: 0.67 9 | width_mult: 0.75 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5p6_m_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | -------------------------------------------------------------------------------- /configs/yolov5/yolov5p6_n_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_high.yml', 5 | '_base_/yolov5p6_cspdarknet.yml', 6 | '_base_/yolov5p6_reader.yml', 7 | ] 8 | depth_mult: 0.33 9 | width_mult: 0.25 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5p6_n_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | -------------------------------------------------------------------------------- /configs/yolov5/yolov5p6_s_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_high.yml', 5 | '_base_/yolov5p6_cspdarknet.yml', 6 | '_base_/yolov5p6_reader.yml', 7 | ] 8 | depth_mult: 0.33 9 | width_mult: 0.50 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5p6_s_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | -------------------------------------------------------------------------------- /configs/yolov5/yolov5p6_x_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_high.yml', 5 | '_base_/yolov5p6_cspdarknet.yml', 6 | '_base_/yolov5p6_reader.yml', 7 | ] 8 | depth_mult: 1.33 9 | width_mult: 1.25 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5p6_x_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 8 # default 8 gpus, total bs = 64 18 | -------------------------------------------------------------------------------- /configs/yolov5u/_base_/optimizer_300e.yml: -------------------------------------------------------------------------------- 1 | epoch: 300 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !YOLOv5LRDecay 7 | max_epochs: 300 8 | min_lr_ratio: 0.01 9 | - !ExpWarmup 10 | epochs: 3 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | type: Momentum 15 | momentum: 0.937 16 | use_nesterov: True 17 | regularizer: 18 | factor: 0.001 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolov5u/_base_/optimizer_300e_high.yml: -------------------------------------------------------------------------------- 1 | epoch: 300 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !YOLOv5LRDecay 7 | max_epochs: 300 8 | min_lr_ratio: 0.1 9 | - !ExpWarmup 10 | epochs: 3 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | type: Momentum 15 | momentum: 0.937 16 | use_nesterov: True 17 | regularizer: 18 | factor: 0.001 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolov5u/_base_/yolov5up6_cspdarknet.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'yolov5u_cspdarknet.yml', 3 | ] 4 | 5 | CSPDarkNet: 6 | arch: "P6" 7 | return_idx: [2, 3, 4, 5] 8 | 9 | YOLOCSPPAN: 10 | depthwise: false 11 | 12 | YOLOv8Head: 13 | fpn_strides: [8, 16, 32, 64] 14 | -------------------------------------------------------------------------------- /configs/yolov5u/yolov5u_l_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_high.yml', 5 | '_base_/yolov5u_cspdarknet.yml', 6 | '_base_/yolov5u_reader_high_aug.yml', 7 | ] 8 | depth_mult: 1.0 9 | width_mult: 1.0 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5u_l_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | -------------------------------------------------------------------------------- /configs/yolov5u/yolov5u_m_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_high.yml', 5 | '_base_/yolov5u_cspdarknet.yml', 6 | '_base_/yolov5u_reader_high_aug.yml', 7 | ] 8 | depth_mult: 0.67 9 | width_mult: 0.75 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5u_m_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | -------------------------------------------------------------------------------- /configs/yolov5u/yolov5u_n_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e.yml', 5 | '_base_/yolov5u_cspdarknet.yml', 6 | '_base_/yolov5u_reader.yml', 7 | ] 8 | depth_mult: 0.33 9 | width_mult: 0.25 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5u_n_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | -------------------------------------------------------------------------------- /configs/yolov5u/yolov5u_s_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e.yml', 5 | '_base_/yolov5u_cspdarknet.yml', 6 | '_base_/yolov5u_reader.yml', 7 | ] 8 | depth_mult: 0.33 9 | width_mult: 0.50 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5u_s_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | -------------------------------------------------------------------------------- /configs/yolov5u/yolov5u_x_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_high.yml', 5 | '_base_/yolov5u_cspdarknet.yml', 6 | '_base_/yolov5u_reader_high_aug.yml', 7 | ] 8 | depth_mult: 1.33 9 | width_mult: 1.25 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5u_x_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | -------------------------------------------------------------------------------- /configs/yolov5u/yolov5up6_l_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_high.yml', 5 | '_base_/yolov5up6_cspdarknet.yml', 6 | '_base_/yolov5up6_reader.yml', 7 | ] 8 | depth_mult: 1.0 9 | width_mult: 1.0 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5up6_l_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 8 # default 8 gpus, total bs = 64 18 | -------------------------------------------------------------------------------- /configs/yolov5u/yolov5up6_m_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_high.yml', 5 | '_base_/yolov5up6_cspdarknet.yml', 6 | '_base_/yolov5up6_reader.yml', 7 | ] 8 | depth_mult: 0.67 9 | width_mult: 0.75 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5up6_m_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | -------------------------------------------------------------------------------- /configs/yolov5u/yolov5up6_n_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e.yml', 5 | '_base_/yolov5up6_cspdarknet.yml', 6 | '_base_/yolov5up6_reader.yml', 7 | ] 8 | depth_mult: 0.33 9 | width_mult: 0.25 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5up6_n_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | -------------------------------------------------------------------------------- /configs/yolov5u/yolov5up6_s_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e.yml', 5 | '_base_/yolov5up6_cspdarknet.yml', 6 | '_base_/yolov5up6_reader.yml', 7 | ] 8 | depth_mult: 0.33 9 | width_mult: 0.50 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5up6_s_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | -------------------------------------------------------------------------------- /configs/yolov5u/yolov5up6_x_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_high.yml', 5 | '_base_/yolov5up6_cspdarknet.yml', 6 | '_base_/yolov5up6_reader.yml', 7 | ] 8 | depth_mult: 1.33 9 | width_mult: 1.25 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolov5up6_x_300e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 8 # default 8 gpus, total bs = 64 18 | -------------------------------------------------------------------------------- /configs/yolov6/_base_/optimizer_300e.yml: -------------------------------------------------------------------------------- 1 | epoch: 300 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !YOLOv5LRDecay 7 | max_epochs: 300 8 | min_lr_ratio: 0.01 9 | - !ExpWarmup 10 | epochs: 3 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | type: Momentum 15 | momentum: 0.937 16 | use_nesterov: True 17 | regularizer: 18 | factor: 0.0005 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolov6/_base_/optimizer_300e_fintune.yml: -------------------------------------------------------------------------------- 1 | epoch: 300 2 | 3 | LearningRate: 4 | base_lr: 0.0032 5 | schedulers: 6 | - !YOLOv5LRDecay 7 | max_epochs: 300 8 | min_lr_ratio: 0.12 9 | - !ExpWarmup 10 | epochs: 2 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | type: Momentum 15 | momentum: 0.843 16 | use_nesterov: True 17 | regularizer: 18 | factor: 0.00036 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolov6/yolov6lite/_base_/optimizer_400e.yml: -------------------------------------------------------------------------------- 1 | epoch: 400 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !YOLOv5LRDecay 7 | max_epochs: 400 8 | min_lr_ratio: 0.01 9 | - !ExpWarmup 10 | epochs: 3 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | type: Momentum 15 | momentum: 0.937 16 | use_nesterov: True 17 | regularizer: 18 | factor: 0.0005 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolov6/yolov6lite/_base_/optimizer_400e_fintune.yml: -------------------------------------------------------------------------------- 1 | epoch: 400 2 | 3 | LearningRate: 4 | base_lr: 0.0032 5 | schedulers: 6 | - !YOLOv5LRDecay 7 | max_epochs: 400 8 | min_lr_ratio: 0.12 9 | - !ExpWarmup 10 | epochs: 2 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | type: Momentum 15 | momentum: 0.843 16 | use_nesterov: True 17 | regularizer: 18 | factor: 0.00036 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolov6/yolov6lite/yolov6lite_l_400e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/coco_detection.yml', 3 | '../../runtime.yml', 4 | '_base_/optimizer_400e.yml', 5 | '_base_/yolov6lite_effibackbone.yml', 6 | '_base_/yolov6lite_reader_320.yml', 7 | ] 8 | width_mult: 1.5 9 | log_iter: 100 10 | snapshot_epoch: 10 11 | weights: output/yolov6lite_l_400e_coco/model_final 12 | 13 | 14 | ### reader config 15 | TrainReader: 16 | batch_size: 32 # default 8 gpus, total bs = 256 17 | 18 | EvalReader: 19 | batch_size: 8 20 | -------------------------------------------------------------------------------- /configs/yolov6/yolov6lite/yolov6lite_m_400e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/coco_detection.yml', 3 | '../../runtime.yml', 4 | '_base_/optimizer_400e.yml', 5 | '_base_/yolov6lite_effibackbone.yml', 6 | '_base_/yolov6lite_reader_320.yml', 7 | ] 8 | width_mult: 1.1 9 | log_iter: 100 10 | snapshot_epoch: 10 11 | weights: output/yolov6lite_m_400e_coco/model_final 12 | 13 | 14 | ### reader config 15 | TrainReader: 16 | batch_size: 32 # default 8 gpus, total bs = 256 17 | 18 | EvalReader: 19 | batch_size: 8 20 | -------------------------------------------------------------------------------- /configs/yolov6/yolov6lite/yolov6lite_s_400e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../../datasets/coco_detection.yml', 3 | '../../runtime.yml', 4 | '_base_/optimizer_400e.yml', 5 | '_base_/yolov6lite_effibackbone.yml', 6 | '_base_/yolov6lite_reader_320.yml', 7 | ] 8 | width_mult: 0.7 9 | log_iter: 100 10 | snapshot_epoch: 10 11 | weights: output/yolov6lite_s_400e_coco/model_final 12 | 13 | 14 | ### reader config 15 | TrainReader: 16 | batch_size: 32 # default 8 gpus, total bs = 256 17 | 18 | EvalReader: 19 | batch_size: 8 20 | -------------------------------------------------------------------------------- /configs/yolov7/_base_/optimizer_300e.yml: -------------------------------------------------------------------------------- 1 | epoch: 300 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !YOLOv5LRDecay 7 | max_epochs: 300 8 | min_lr_ratio: 0.1 9 | - !ExpWarmup 10 | epochs: 3 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | type: Momentum 15 | momentum: 0.937 16 | use_nesterov: True 17 | regularizer: 18 | factor: 0.0005 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolov7/_base_/optimizer_300e_p6.yml: -------------------------------------------------------------------------------- 1 | epoch: 300 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !YOLOv5LRDecay 7 | max_epochs: 300 8 | min_lr_ratio: 0.2 # 9 | - !ExpWarmup 10 | epochs: 3 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | type: Momentum 15 | momentum: 0.937 16 | use_nesterov: True 17 | regularizer: 18 | factor: 0.0005 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolov7/_base_/optimizer_300e_tiny.yml: -------------------------------------------------------------------------------- 1 | epoch: 300 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !YOLOv5LRDecay 7 | max_epochs: 300 8 | min_lr_ratio: 0.01 # 9 | - !ExpWarmup 10 | epochs: 3 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | type: Momentum 15 | momentum: 0.937 16 | use_nesterov: True 17 | regularizer: 18 | factor: 0.0005 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolov7/yolov7_l_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e.yml', 5 | '_base_/yolov7_elannet.yml', 6 | '_base_/yolov7_reader.yml', 7 | ] 8 | log_iter: 100 9 | snapshot_epoch: 10 10 | weights: output/yolov7_l_300e_coco/model_final 11 | 12 | arch: L 13 | 14 | 15 | TrainReader: 16 | batch_size: 32 17 | -------------------------------------------------------------------------------- /configs/yolov7/yolov7_tiny_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_tiny.yml', 5 | '_base_/yolov7_elannet.yml', 6 | '_base_/yolov7_tiny_reader.yml', 7 | ] 8 | log_iter: 100 9 | snapshot_epoch: 10 10 | weights: output/yolov7_tiny_300e_coco/model_final 11 | 12 | arch: tiny 13 | act: LeakyReLU 14 | 15 | 16 | TrainReader: 17 | batch_size: 32 18 | 19 | 20 | YOLOv7Loss: 21 | box_weight: 0.05 22 | cls_weght: 0.5 23 | obj_weight: 1.0 24 | -------------------------------------------------------------------------------- /configs/yolov7/yolov7_x_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e.yml', 5 | '_base_/yolov7_elannet.yml', 6 | '_base_/yolov7_reader.yml', 7 | ] 8 | log_iter: 100 9 | snapshot_epoch: 10 10 | weights: output/yolov7_x_300e_coco/model_final 11 | 12 | arch: X 13 | 14 | 15 | TrainReader: 16 | batch_size: 32 17 | -------------------------------------------------------------------------------- /configs/yolov7/yolov7p6_d6_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_p6.yml', 5 | '_base_/yolov7p6_elannet.yml', 6 | '_base_/yolov7p6_reader.yml', 7 | ] 8 | log_iter: 100 9 | snapshot_epoch: 10 10 | weights: output/yolov7p6_d6_300e_coco/model_final 11 | 12 | arch: D6 13 | use_aux: True 14 | 15 | 16 | TrainReader: 17 | batch_size: 8 18 | -------------------------------------------------------------------------------- /configs/yolov7/yolov7p6_e6_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_p6.yml', 5 | '_base_/yolov7p6_elannet.yml', 6 | '_base_/yolov7p6_reader.yml', 7 | ] 8 | log_iter: 100 9 | snapshot_epoch: 10 10 | weights: output/yolov7p6_e6_300e_coco/model_final 11 | 12 | arch: E6 13 | use_aux: True 14 | 15 | 16 | TrainReader: 17 | batch_size: 10 18 | -------------------------------------------------------------------------------- /configs/yolov7/yolov7p6_e6e_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_p6.yml', 5 | '_base_/yolov7p6_elannet.yml', 6 | '_base_/yolov7p6_reader.yml', 7 | ] 8 | log_iter: 100 9 | snapshot_epoch: 10 10 | weights: output/yolov7p6_e6e_300e_coco/model_final 11 | 12 | arch: E6E 13 | use_aux: True 14 | 15 | 16 | TrainReader: 17 | batch_size: 6 18 | -------------------------------------------------------------------------------- /configs/yolov7/yolov7p6_w6_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e_p6.yml', 5 | '_base_/yolov7p6_elannet.yml', 6 | '_base_/yolov7p6_reader.yml', 7 | ] 8 | log_iter: 100 9 | snapshot_epoch: 10 10 | weights: output/yolov7p6_w6_300e_coco/model_final 11 | 12 | arch: W6 13 | use_aux: True 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 18 | -------------------------------------------------------------------------------- /configs/yolov7u/_base_/optimizer_300e.yml: -------------------------------------------------------------------------------- 1 | epoch: 300 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !YOLOv5LRDecay 7 | max_epochs: 300 8 | min_lr_ratio: 0.1 9 | - !ExpWarmup 10 | epochs: 3 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | type: Momentum 15 | momentum: 0.937 16 | use_nesterov: True 17 | regularizer: 18 | factor: 0.001 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolov7u/yolov7u_l_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_300e.yml', 5 | '_base_/yolov7u_elannet.yml', 6 | '_base_/yolov7u_reader.yml', 7 | ] 8 | log_iter: 100 9 | snapshot_epoch: 10 10 | weights: output/yolov7u_l_300e_coco/model_final 11 | 12 | arch: L 13 | 14 | 15 | TrainReader: 16 | batch_size: 8 #16 # default 8 gpus, total bs = 128 17 | -------------------------------------------------------------------------------- /configs/yolov8/_base_/optimizer_100e.yml: -------------------------------------------------------------------------------- 1 | epoch: 100 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !YOLOv5LRDecay 7 | max_epochs: 100 8 | min_lr_ratio: 0.01 9 | - !ExpWarmup 10 | epochs: 3 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | type: Momentum 15 | momentum: 0.937 16 | use_nesterov: True 17 | regularizer: 18 | factor: 0.0005 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolov8/_base_/optimizer_300e.yml: -------------------------------------------------------------------------------- 1 | epoch: 300 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !YOLOv5LRDecay 7 | max_epochs: 300 8 | min_lr_ratio: 0.01 9 | - !ExpWarmup 10 | epochs: 3 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | type: Momentum 15 | momentum: 0.937 16 | use_nesterov: True 17 | regularizer: 18 | factor: 0.0005 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolov8/_base_/optimizer_500e.yml: -------------------------------------------------------------------------------- 1 | epoch: 500 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !YOLOv5LRDecay 7 | max_epochs: 500 8 | min_lr_ratio: 0.01 9 | - !ExpWarmup 10 | epochs: 3 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | type: Momentum 15 | momentum: 0.937 16 | use_nesterov: True 17 | regularizer: 18 | factor: 0.0005 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolov8/_base_/optimizer_500e_high.yml: -------------------------------------------------------------------------------- 1 | epoch: 500 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !YOLOv5LRDecay 7 | max_epochs: 500 8 | min_lr_ratio: 0.1 # 9 | - !ExpWarmup 10 | epochs: 3 11 | 12 | OptimizerBuilder: 13 | optimizer: 14 | type: Momentum 15 | momentum: 0.937 16 | use_nesterov: True 17 | regularizer: 18 | factor: 0.0005 19 | type: L2 20 | -------------------------------------------------------------------------------- /configs/yolov8/yolov8_l_500e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_500e_high.yml', 5 | '_base_/yolov8_cspdarknet.yml', 6 | '_base_/yolov8_reader_high_aug.yml', 7 | ] 8 | depth_mult: 1.0 9 | width_mult: 1.0 10 | 11 | log_iter: 50 12 | snapshot_epoch: 10 13 | weights: output/yolov8_l_500e_coco/model_final 14 | 15 | 16 | YOLOv8CSPDarkNet: 17 | last_stage_ch: 512 # The actual channel is int(512 * width_mult), not int(1024 * width_mult) as in YOLOv5 18 | 19 | 20 | TrainReader: 21 | batch_size: 16 # default 8 gpus, total bs = 128 22 | -------------------------------------------------------------------------------- /configs/yolov8/yolov8_m_500e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_500e_high.yml', 5 | '_base_/yolov8_cspdarknet.yml', 6 | '_base_/yolov8_reader_high_aug.yml', 7 | ] 8 | depth_mult: 0.67 9 | width_mult: 0.75 10 | 11 | log_iter: 50 12 | snapshot_epoch: 10 13 | weights: output/yolov8_m_500e_coco/model_final 14 | 15 | 16 | YOLOv8CSPDarkNet: 17 | last_stage_ch: 768 # The actual channel is int(768 * width_mult), not int(1024 * width_mult) as in YOLOv5 18 | 19 | 20 | TrainReader: 21 | batch_size: 16 # default 8 gpus, total bs = 128 22 | -------------------------------------------------------------------------------- /configs/yolov8/yolov8_n_500e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_500e.yml', 5 | '_base_/yolov8_cspdarknet.yml', 6 | '_base_/yolov8_reader.yml', 7 | ] 8 | depth_mult: 0.33 9 | width_mult: 0.25 10 | 11 | log_iter: 50 12 | snapshot_epoch: 10 13 | weights: output/yolov8_n_500e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | -------------------------------------------------------------------------------- /configs/yolov8/yolov8_s_500e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_500e.yml', 5 | '_base_/yolov8_cspdarknet.yml', 6 | '_base_/yolov8_reader.yml', 7 | ] 8 | depth_mult: 0.33 9 | width_mult: 0.50 10 | 11 | log_iter: 50 12 | snapshot_epoch: 10 13 | weights: output/yolov8_s_500e_coco/model_final 14 | 15 | 16 | TrainReader: 17 | batch_size: 16 # default 8 gpus, total bs = 128 18 | -------------------------------------------------------------------------------- /configs/yolov8/yolov8_x_500e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | '_base_/optimizer_500e_high.yml', 5 | '_base_/yolov8_cspdarknet.yml', 6 | '_base_/yolov8_reader_high_aug.yml', 7 | ] 8 | depth_mult: 1.0 # not 1.33 as in YOLOv5 9 | width_mult: 1.25 10 | 11 | log_iter: 50 12 | snapshot_epoch: 10 13 | weights: output/yolov8_x_500e_coco/model_final 14 | 15 | 16 | YOLOv8CSPDarkNet: 17 | last_stage_ch: 512 # The actual channel is int(512 * width_mult), not int(1024 * width_mult) as in YOLOv5 18 | 19 | 20 | TrainReader: 21 | batch_size: 16 # default 8 gpus, total bs = 128 22 | -------------------------------------------------------------------------------- /configs/yolox/_base_/optimizer_300e.yml: -------------------------------------------------------------------------------- 1 | epoch: 300 2 | 3 | LearningRate: 4 | base_lr: 0.01 5 | schedulers: 6 | - !CosineDecay 7 | max_epochs: 300 8 | min_lr_ratio: 0.05 9 | last_plateau_epochs: 15 10 | - !ExpWarmup 11 | epochs: 5 12 | 13 | OptimizerBuilder: 14 | optimizer: 15 | type: Momentum 16 | momentum: 0.9 17 | use_nesterov: True 18 | regularizer: 19 | factor: 0.0005 20 | type: L2 21 | -------------------------------------------------------------------------------- /configs/yolox/yolox_cdn_tiny_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | 'yolox_tiny_300e_coco.yml' 3 | ] 4 | depth_mult: 0.33 5 | width_mult: 0.375 6 | 7 | log_iter: 100 8 | snapshot_epoch: 10 9 | weights: output/yolox_cdn_tiny_300e_coco/model_final 10 | 11 | CSPDarkNet: 12 | arch: "P5" # using the same backbone of YOLOv5 releases v6.0 and later version 13 | return_idx: [2, 3, 4] 14 | depthwise: False 15 | -------------------------------------------------------------------------------- /configs/yolox/yolox_l_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/yolox_cspdarknet.yml', 6 | './_base_/yolox_reader.yml' 7 | ] 8 | depth_mult: 1.0 9 | width_mult: 1.0 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolox_l_300e_coco/model_final 14 | -------------------------------------------------------------------------------- /configs/yolox/yolox_m_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/yolox_cspdarknet.yml', 6 | './_base_/yolox_reader.yml' 7 | ] 8 | depth_mult: 0.67 9 | width_mult: 0.75 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolox_m_300e_coco/model_final 14 | -------------------------------------------------------------------------------- /configs/yolox/yolox_s_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/yolox_cspdarknet.yml', 6 | './_base_/yolox_reader.yml' 7 | ] 8 | depth_mult: 0.33 9 | width_mult: 0.50 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolox_s_300e_coco/model_final 14 | -------------------------------------------------------------------------------- /configs/yolox/yolox_x_300e_coco.yml: -------------------------------------------------------------------------------- 1 | _BASE_: [ 2 | '../datasets/coco_detection.yml', 3 | '../runtime.yml', 4 | './_base_/optimizer_300e.yml', 5 | './_base_/yolox_cspdarknet.yml', 6 | './_base_/yolox_reader.yml' 7 | ] 8 | depth_mult: 1.33 9 | width_mult: 1.25 10 | 11 | log_iter: 100 12 | snapshot_epoch: 10 13 | weights: output/yolox_x_300e_coco/model_final 14 | -------------------------------------------------------------------------------- /demo/000000014439.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/demo/000000014439.jpg -------------------------------------------------------------------------------- /demo/000000014439_640x640.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/demo/000000014439_640x640.jpg -------------------------------------------------------------------------------- /demo/000000087038.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/demo/000000087038.jpg -------------------------------------------------------------------------------- /demo/000000570688.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/demo/000000570688.jpg -------------------------------------------------------------------------------- /demo/0000315_01601_d_0000509.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/demo/0000315_01601_d_0000509.jpg -------------------------------------------------------------------------------- /demo/39006.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/demo/39006.jpg -------------------------------------------------------------------------------- /demo/P0072__1.0__0___0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/demo/P0072__1.0__0___0.png -------------------------------------------------------------------------------- /demo/P0861__1.0__1154___824.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/demo/P0861__1.0__1154___824.png -------------------------------------------------------------------------------- /demo/car.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/demo/car.jpg -------------------------------------------------------------------------------- /demo/hrnet_demo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/demo/hrnet_demo.jpg -------------------------------------------------------------------------------- /demo/orange_71.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/demo/orange_71.jpg -------------------------------------------------------------------------------- /demo/road554.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/demo/road554.png -------------------------------------------------------------------------------- /docs/advanced_tutorials/openvino_inference/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | opencv-python 3 | openvino == 2021.4.0 4 | paddledet==2.3.0 -------------------------------------------------------------------------------- /docs/advanced_tutorials/openvino_inference/street.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/advanced_tutorials/openvino_inference/street.jpeg -------------------------------------------------------------------------------- /docs/advanced_tutorials/openvino_inference/tracked.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/advanced_tutorials/openvino_inference/tracked.jpg -------------------------------------------------------------------------------- /docs/images/000000014439.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/000000014439.jpg -------------------------------------------------------------------------------- /docs/images/12_Group_Group_12_Group_Group_12_935.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/12_Group_Group_12_Group_Group_12_935.jpg -------------------------------------------------------------------------------- /docs/images/PedestrianDetection_001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/PedestrianDetection_001.png -------------------------------------------------------------------------------- /docs/images/PedestrianDetection_004.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/PedestrianDetection_004.png -------------------------------------------------------------------------------- /docs/images/VehicleDetection_001.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/VehicleDetection_001.jpeg -------------------------------------------------------------------------------- /docs/images/VehicleDetection_005.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/VehicleDetection_005.png -------------------------------------------------------------------------------- /docs/images/add_attribute.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/add_attribute.png -------------------------------------------------------------------------------- /docs/images/bus.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/bus.jpg -------------------------------------------------------------------------------- /docs/images/dog.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/dog.jpg -------------------------------------------------------------------------------- /docs/images/fps_map.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/fps_map.png -------------------------------------------------------------------------------- /docs/images/grad_cam_ppyoloe_demo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/grad_cam_ppyoloe_demo.jpg -------------------------------------------------------------------------------- /docs/images/input_shape.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/input_shape.png -------------------------------------------------------------------------------- /docs/images/instance_seg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/instance_seg.png -------------------------------------------------------------------------------- /docs/images/layout.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/layout.jpg -------------------------------------------------------------------------------- /docs/images/layout_demo1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/layout_demo1.png -------------------------------------------------------------------------------- /docs/images/layout_demo2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/layout_demo2.png -------------------------------------------------------------------------------- /docs/images/lite_demo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/lite_demo.jpg -------------------------------------------------------------------------------- /docs/images/mobile_fps_map.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/mobile_fps_map.png -------------------------------------------------------------------------------- /docs/images/model_figure.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/model_figure.png -------------------------------------------------------------------------------- /docs/images/picedet_demo.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/picedet_demo.jpeg -------------------------------------------------------------------------------- /docs/images/picodet_android_demo1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/picodet_android_demo1.jpg -------------------------------------------------------------------------------- /docs/images/picodet_android_demo2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/picodet_android_demo2.jpg -------------------------------------------------------------------------------- /docs/images/picodet_android_demo3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/picodet_android_demo3.jpg -------------------------------------------------------------------------------- /docs/images/picodet_android_demo4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/picodet_android_demo4.jpg -------------------------------------------------------------------------------- /docs/images/picodet_map.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/picodet_map.png -------------------------------------------------------------------------------- /docs/images/pphuman-tech.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/pphuman-tech.png -------------------------------------------------------------------------------- /docs/images/pphumanv2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/pphumanv2.png -------------------------------------------------------------------------------- /docs/images/pptracking.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/pptracking.png -------------------------------------------------------------------------------- /docs/images/pptracking_en.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/pptracking_en.png -------------------------------------------------------------------------------- /docs/images/ppvehicle.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/ppvehicle.png -------------------------------------------------------------------------------- /docs/images/ppyolo_map_fps.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/ppyolo_map_fps.png -------------------------------------------------------------------------------- /docs/images/ppyoloe_plus_map_fps.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/ppyoloe_plus_map_fps.png -------------------------------------------------------------------------------- /docs/images/ppyoloe_r_map_fps.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/ppyoloe_r_map_fps.png -------------------------------------------------------------------------------- /docs/images/reader_figure.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/reader_figure.png -------------------------------------------------------------------------------- /docs/images/res.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/res.jpg -------------------------------------------------------------------------------- /docs/images/road554.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/road554.png -------------------------------------------------------------------------------- /docs/images/roadsign_yml.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/roadsign_yml.png -------------------------------------------------------------------------------- /docs/images/ssld_model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/ssld_model.png -------------------------------------------------------------------------------- /docs/images/tinypose_app.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/tinypose_app.png -------------------------------------------------------------------------------- /docs/images/tinypose_demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/tinypose_demo.png -------------------------------------------------------------------------------- /docs/images/tinypose_pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/tinypose_pipeline.png -------------------------------------------------------------------------------- /docs/images/yaml_show.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/docs/images/yaml_show.png -------------------------------------------------------------------------------- /docs/tutorials/FAQ/README.md: -------------------------------------------------------------------------------- 1 | # FAQ/常见问题 2 | 3 | **PaddleDetection**非常感谢各位开发者提出任何使用问题或需求，我们根据大家的提问，总结**FAQ/常见问题**合集，并在**每周一**进行更新，以下是往期的FAQ，欢迎大家进行查阅。 4 | 5 | - [FAQ：第零期](./FAQ第零期.md) 6 | - [FAQ：第一期](./FAQ第一期.md) 7 | -------------------------------------------------------------------------------- /ppdettorch/__init__.py: -------------------------------------------------------------------------------- 1 | #!/user/bin/env python 2 | # -*- coding: utf-8 -*- 3 | # @File : 4 | # @Author: sl 5 | # @Date : 2022/12/19 - 22:14 6 | -------------------------------------------------------------------------------- /ppdettorch/core/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: UTF-8 -*- 3 | # @Project ：PPDetectionPytorch 4 | # @File ：__init__.py.py 5 | # @Author ：sl 6 | # @Date ：2022/11/1 14:04 7 | 8 | 9 | from . import config 10 | 11 | -------------------------------------------------------------------------------- /ppdettorch/core/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/ppdettorch/core/config/__init__.py -------------------------------------------------------------------------------- /ppdettorch/data/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: UTF-8 -*- 3 | # @Project ：PPDetectionPytorch 4 | # @File ：__init__.py.py 5 | # @Author ：sl 6 | # @Date ：2022/11/3 15:47 7 | 8 | from . import source 9 | from . import transform 10 | from . import reader 11 | 12 | from .source import * 13 | from .transform import * 14 | from .reader import * 15 | -------------------------------------------------------------------------------- /ppdettorch/data/crop_utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved. 2 | # 3 | # Licensed under the Apache License, Version 2.0 (the "License"); 4 | # you may not use this file except in compliance with the License. 5 | # You may obtain a copy of the License at 6 | # 7 | # http://www.apache.org/licenses/LICENSE-2.0 8 | # 9 | # Unless required by applicable law or agreed to in writing, software 10 | # distributed under the License is distributed on an "AS IS" BASIS, 11 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 | # See the License for the specific language governing permissions and 13 | # limitations under the License. -------------------------------------------------------------------------------- /ppdettorch/data/source/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: UTF-8 -*- 3 | # @Project ：PPDetectionPytorch 4 | # @File ：__init__.py.py 5 | # @Author ：sl 6 | # @Date ：2022/11/3 16:27 7 | 8 | from . import coco 9 | from . import voc 10 | from . import widerface 11 | from . import category 12 | from . import keypoint_coco 13 | from . import mot 14 | from . import sniper_coco 15 | from . import dataset 16 | 17 | from .coco import * 18 | from .voc import * 19 | from .widerface import * 20 | from .category import * 21 | from .keypoint_coco import * 22 | from .mot import * 23 | from .sniper_coco import SniperCOCODataSet 24 | from .dataset import * 25 | -------------------------------------------------------------------------------- /ppdettorch/engine/__init__.py: -------------------------------------------------------------------------------- 1 | #!/user/bin/env python 2 | # -*- coding: utf-8 -*- 3 | # @File : 4 | # @Author: sl 5 | # @Date : 2022/12/20 - 22:16 6 | -------------------------------------------------------------------------------- /ppdettorch/metrics/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: UTF-8 -*- 3 | # @Project ：PPDetectionPytorch 4 | # @File ：__init__.py.py 5 | # @Author ：sl 6 | # @Date ：2022/11/7 17:55 7 | 8 | from . import metrics 9 | from . import keypoint_metrics 10 | 11 | from .metrics import * 12 | from .keypoint_metrics import * 13 | 14 | __all__ = metrics.__all__ + keypoint_metrics.__all__ 15 | 16 | from . import mot_metrics 17 | from .mot_metrics import * 18 | 19 | __all__ = metrics.__all__ + mot_metrics.__all__ 20 | 21 | from . import mcmot_metrics 22 | from .mcmot_metrics import * 23 | 24 | __all__ = metrics.__all__ + mcmot_metrics.__all__ 25 | -------------------------------------------------------------------------------- /ppdettorch/modeling/assigners/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: UTF-8 -*- 3 | # @Project ：PPDetectionPytorch 4 | # @File ：__init__.py.py 5 | # @Author ：sl 6 | # @Date ：2022/11/1 18:54 7 | 8 | from . import utils 9 | from . import task_aligned_assigner 10 | from . import atss_assigner 11 | from . import simota_assigner 12 | from . import max_iou_assigner 13 | 14 | from .utils import * 15 | from .task_aligned_assigner import * 16 | from .atss_assigner import * 17 | from .simota_assigner import * 18 | from .max_iou_assigner import * 19 | -------------------------------------------------------------------------------- /ppdettorch/modeling/proposal_generator/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: UTF-8 -*- 3 | # @Project ：PPDetectionPytorch 4 | # @File ：__init__.py.py 5 | # @Author ：sl 6 | # @Date ：2022/11/4 15:05 7 | 8 | from . import rpn_head 9 | from .rpn_head import * 10 | -------------------------------------------------------------------------------- /ppdettorch/modeling/tests/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved. 2 | # 3 | # Licensed under the Apache License, Version 2.0 (the "License"); 4 | # you may not use this file except in compliance with the License. 5 | # You may obtain a copy of the License at 6 | # 7 | # http://www.apache.org/licenses/LICENSE-2.0 8 | # 9 | # Unless required by applicable law or agreed to in writing, software 10 | # distributed under the License is distributed on an "AS IS" BASIS, 11 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 | # See the License for the specific language governing permissions and 13 | # limitations under the License. 14 | -------------------------------------------------------------------------------- /ppdettorch/modeling/tests/imgs/coco2017_val2017_000000000139.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/ppdettorch/modeling/tests/imgs/coco2017_val2017_000000000139.jpg -------------------------------------------------------------------------------- /ppdettorch/modeling/tests/imgs/coco2017_val2017_000000000724.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CycloneBoy/PPDetectionPytorch/124eb186fbe72d4165485fa07cae16bd7ed7d532/ppdettorch/modeling/tests/imgs/coco2017_val2017_000000000724.jpg -------------------------------------------------------------------------------- /ppdettorch/optimizer/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: UTF-8 -*- 3 | # @Project ：PPDetectionPytorch 4 | # @File ：__init__.py.py 5 | # @Author ：sl 6 | # @Date ：2022/11/3 17:28 7 | 8 | from .optimizer import * 9 | from .ema import ModelEMA -------------------------------------------------------------------------------- /ppdettorch/process/__init__.py: -------------------------------------------------------------------------------- 1 | #!/user/bin/env python 2 | # -*- coding: utf-8 -*- 3 | # @File : 4 | # @Author: sl 5 | # @Date : 2022/12/20 - 22:04 6 | -------------------------------------------------------------------------------- /ppdettorch/process/infer/__init__.py: -------------------------------------------------------------------------------- 1 | #!/user/bin/env python 2 | # -*- coding: utf-8 -*- 3 | # @File : 4 | # @Author: sl 5 | # @Date : 2022/12/21 - 22:00 6 | -------------------------------------------------------------------------------- /ppdettorch/process/transform/__init__.py: -------------------------------------------------------------------------------- 1 | #!/user/bin/env python 2 | # -*- coding: utf-8 -*- 3 | # @File : 4 | # @Author: sl 5 | # @Date : 2022/12/20 - 22:05 6 | -------------------------------------------------------------------------------- /ppdettorch/utils/__init__.py: -------------------------------------------------------------------------------- 1 | #!/user/bin/env python 2 | # -*- coding: utf-8 -*- 3 | # @File : 4 | # @Author: sl 5 | # @Date : 2022/12/20 - 21:44 6 | -------------------------------------------------------------------------------- /ppdettorch/utils/base_utils.py: -------------------------------------------------------------------------------- 1 | #!/user/bin/env python 2 | # -*- coding: utf-8 -*- 3 | # @File : 4 | # @Author: sl 5 | # @Date : 2022/12/20 - 21:58 6 | import os 7 | from abc import abstractmethod, ABC 8 | from enum import Enum, unique 9 | 10 | 11 | class BaseUtil(ABC): 12 | """ 13 | 抽取数据基类 14 | """ 15 | 16 | @abstractmethod 17 | def init(self): 18 | """ 19 | 工具类初始化 20 | :return: 21 | """ 22 | pass 23 | -------------------------------------------------------------------------------- /ppdettorch/utils/constant.py: -------------------------------------------------------------------------------- 1 | #!/user/bin/env python 2 | # -*- coding: utf-8 -*- 3 | # @File : 4 | # @Author: sl 5 | # @Date : 2022/12/20 - 21:56 6 | import os 7 | import re 8 | 9 | 10 | class Constants(object): 11 | """ 12 | 常量工具类 13 | 14 | """ 15 | USER_HOME = os.path.expanduser("~") 16 | WORK_DIR = f"{USER_HOME}/work/PPDetectionPytorch" 17 | # WORK_DIR = "./" 18 | 19 | DATA_DIR = f"{WORK_DIR}/data" 20 | LOG_DIR = f"{WORK_DIR}/logs" 21 | 22 | # 日志相关 23 | LOG_FILE = f"{LOG_DIR}/run.log" 24 | LOG_LEVEL = "debug" 25 | 26 | # models 27 | OUTPUT_MODELS_DIR = f"{WORK_DIR}/outputs/models" 28 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | tqdm 2 | typeguard 3 | visualdl>=2.2.0 4 | opencv-python 5 | PyYAML 6 | shapely 7 | scipy 8 | terminaltables 9 | Cython 10 | pycocotools 11 | setuptools 12 | 13 | # for vehicleplate 14 | pyclipper 15 | 16 | # for mot 17 | lap 18 | motmetrics 19 | sklearn 20 | filterpy 21 | # paddlepaddle-gpu 22 | torch 23 | -------------------------------------------------------------------------------- /tests/process/__init__.py: -------------------------------------------------------------------------------- 1 | #!/user/bin/env python 2 | # -*- coding: utf-8 -*- 3 | # @File : 4 | # @Author: sl 5 | # @Date : 2022/12/21 - 21:48 6 | -------------------------------------------------------------------------------- /tests/process/infer/__init__.py: -------------------------------------------------------------------------------- 1 | #!/user/bin/env python 2 | # -*- coding: utf-8 -*- 3 | # @File : 4 | # @Author: sl 5 | # @Date : 2022/12/21 - 21:48 6 | --------------------------------------------------------------------------------