leo/khqp_trainer
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Files
1077116a8b05e69b524f0384c11a76ab35941418
khqp_trainer/bak/evolve_config2.py
leo 31a2a947b4 flush db v1
2024-03-06 20:30:27 +08:00

494 lines
23 KiB
Python
Raw Blame History

update_cache = False
train_data = [
# khdp_7flys
{
'img_prefix': '/data/data/split_ss_yimu_m6_p2/train_khdp_7flys_gsd5_orgsize_noDecay_seg-v2/images',
'ann_file': '/data/data/split_ss_yimu_m6_p2/train_khdp_7flys_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230531_p1p2',
'select_ids': None, 'filter_empty_gt': False, 'update_cache': False
},
# khdp_v2_3flys
{
'img_prefix': '/data/data/split_ss_yimu_m6_p1/train_khdp_v2_3flys_gsd5_orgsize_noDecay_seg-v2/images',
'ann_file': '/data/data/split_ss_yimu_m6_p1/train_khdp_v2_3flys_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230531_p1p2',
'select_ids': None, 'filter_empty_gt': False, 'update_cache': False
},
# khdp_v3
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_044_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_044_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_064_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_064_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_065_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_065_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_073_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_073_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_072_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_072_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
# khdp_v4
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_013_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_013_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_015_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_015_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_016_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_016_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_028_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_028_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
# khdp_202306_zhubin
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_067_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_067_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_068_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_068_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_069_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_069_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_070_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_070_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_071_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_071_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_075_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_075_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_078_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_078_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_080_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_080_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_081_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_081_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_082_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_082_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_084_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_084_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_087_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_087_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_089_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_089_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_090_mod_loc_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_202306016',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_090_mod_loc_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False, 'update_cache': update_cache
},
# KHXP_DZ
{
'ann_file': '/data/data/split_ss_yimu_khxpdz/patch_2023_khdz_001a002_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khxpdz/patch_2023_khdz_001a002_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khxpdz/patch_2023_khdz_003a004a006_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khxpdz/patch_2023_khdz_003a004a006_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khxpdz/patch_2023_khdz_005_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khxpdz/patch_2023_khdz_005_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khxpdz/patch_2023_khdz_007a008_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khxpdz/patch_2023_khdz_007a008_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
################### khqp 2023 ###################
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_001a006a010_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_001a006a010_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_002a004_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_002a004_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_003_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_003_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_005_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_005_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_007a008a009_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_007a008a009_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_011_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_011_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_012a013_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_012a013_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_014a038_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_014a038_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_015a016_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_015a016_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_018a069_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_018a069_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_019a075_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_019a075_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_020a022_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_020a022_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_023a025_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_023a025_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_026a029a079_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_026a029a079_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_028a027a030_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_028a027a030_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_032a033_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_032a033_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_039a040_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_039a040_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_042a045_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_042a045_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_046a049_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_046a049_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_050a058_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_050a058_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_053a059a060_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_053a059a060_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_054a056_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_054a056_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_063a143_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_063a143_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_073a074a080_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_073a074a080_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_072a081a082_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_072a081a082_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_076_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_076_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_077a078_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_077a078_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_091a097a102a104_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_091a097a102a104_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_087a088a089a090_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_087a088a089a090_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_094_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_094_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
# unqualified data
# {
# 'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_031a101_gsd5_s2048/annfiles',
# 'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_031a101_gsd5_s2048/images',
# 'filter_empty_gt': False,
# 'update_cache': update_cache
# },
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_105a0111_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_105a0111_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_112a113_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_112a113_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_114a116a118_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_114a116a118_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_115_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_115_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_117_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_117_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_119a120_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_119a120_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_034a131_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_034a131_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_132a133_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_132a133_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_057_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_057_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_066a142_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_066a142_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_106a107a138a139_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_106a107a138a139_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
}
]
val_data = [
### test_and_val ###
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_059_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230531',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_059_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False,
'update_cache': update_cache,
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_039A_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230531',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_039A_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False,
'update_cache': update_cache,
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_039B_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230531',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_039B_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False,
'update_cache': update_cache,
},
{
'ann_file': '/data/data/split_ss_yimu_m6_p4/test_khdp_007_gsd5_orgsize_noDecay_seg-v2/annfiles2_labelme_20230601',
'img_prefix': '/data/data/split_ss_yimu_m6_p4/test_khdp_007_gsd5_orgsize_noDecay_seg-v2/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
# khqp 2023
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_095a096a103_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_095a096a103_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_119a120_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_119a120_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
{
'ann_file': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_122_gsd5_s2048/annfiles',
'img_prefix': '/data/data/split_ss_yimu_khqp_2023/patch_2023_khqp_122_gsd5_s2048/images',
'filter_empty_gt': False,
'update_cache': update_cache
},
]
evolve_r = 0.02
ckpt_path = '/data/output/yolox_swinT_small_ds8x_s2048_s3_experiment42/backup_ckpt_mAP0.7649526596069336.pth'
n_trail = 10
n_epoch = 1
Reference in New Issue View Git Blame Copy Permalink