@@ -1189,8 +1189,9 @@ def _recursive_parse_xml_to_dict(xml):
1189
1189
imgs_file_list = load_file_list (path = folder_imgs , regx = '\\ .jpg' , printable = False )
1190
1190
logging .info ("[VOC] {} images found" .format (len (imgs_file_list )))
1191
1191
1192
- imgs_file_list .sort (key = lambda s : int (s .replace ('.' , ' ' ).replace ('_' , '' ).split (' ' )[- 2 ])
1193
- ) # 2007_000027.jpg --> 2007000027
1192
+ imgs_file_list .sort (
1193
+ key = lambda s : int (s .replace ('.' , ' ' ).replace ('_' , '' ).split (' ' )[- 2 ])
1194
+ ) # 2007_000027.jpg --> 2007000027
1194
1195
1195
1196
imgs_file_list = [os .path .join (folder_imgs , s ) for s in imgs_file_list ]
1196
1197
# logging.info('IM',imgs_file_list[0::3333], imgs_file_list[-1])
@@ -1200,17 +1201,19 @@ def _recursive_parse_xml_to_dict(xml):
1200
1201
folder_semseg = os .path .join (path , extracted_filename , "SegmentationClass" )
1201
1202
imgs_semseg_file_list = load_file_list (path = folder_semseg , regx = '\\ .png' , printable = False )
1202
1203
logging .info ("[VOC] {} maps for semantic segmentation found" .format (len (imgs_semseg_file_list )))
1203
- imgs_semseg_file_list .sort (key = lambda s : int (s .replace ('.' , ' ' ).replace ('_' , '' ).split (' ' )[- 2 ])
1204
- ) # 2007_000032.png --> 2007000032
1204
+ imgs_semseg_file_list .sort (
1205
+ key = lambda s : int (s .replace ('.' , ' ' ).replace ('_' , '' ).split (' ' )[- 2 ])
1206
+ ) # 2007_000032.png --> 2007000032
1205
1207
imgs_semseg_file_list = [os .path .join (folder_semseg , s ) for s in imgs_semseg_file_list ]
1206
1208
# logging.info('Semantic Seg IM',imgs_semseg_file_list[0::333], imgs_semseg_file_list[-1])
1207
1209
# ======== 3. instance segmentation maps path list
1208
1210
# folder_insseg = path+"/"+extracted_filename+"/SegmentationObject/"
1209
1211
folder_insseg = os .path .join (path , extracted_filename , "SegmentationObject" )
1210
1212
imgs_insseg_file_list = load_file_list (path = folder_insseg , regx = '\\ .png' , printable = False )
1211
1213
logging .info ("[VOC] {} maps for instance segmentation found" .format (len (imgs_semseg_file_list )))
1212
- imgs_insseg_file_list .sort (key = lambda s : int (s .replace ('.' , ' ' ).replace ('_' , '' ).split (' ' )[- 2 ])
1213
- ) # 2007_000032.png --> 2007000032
1214
+ imgs_insseg_file_list .sort (
1215
+ key = lambda s : int (s .replace ('.' , ' ' ).replace ('_' , '' ).split (' ' )[- 2 ])
1216
+ ) # 2007_000032.png --> 2007000032
1214
1217
imgs_insseg_file_list = [os .path .join (folder_insseg , s ) for s in imgs_insseg_file_list ]
1215
1218
# logging.info('Instance Seg IM',imgs_insseg_file_list[0::333], imgs_insseg_file_list[-1])
1216
1219
else :
@@ -1223,8 +1226,9 @@ def _recursive_parse_xml_to_dict(xml):
1223
1226
logging .info (
1224
1227
"[VOC] {} XML annotation files for bounding box and object class found" .format (len (imgs_ann_file_list ))
1225
1228
)
1226
- imgs_ann_file_list .sort (key = lambda s : int (s .replace ('.' , ' ' ).replace ('_' , '' ).split (' ' )[- 2 ])
1227
- ) # 2007_000027.xml --> 2007000027
1229
+ imgs_ann_file_list .sort (
1230
+ key = lambda s : int (s .replace ('.' , ' ' ).replace ('_' , '' ).split (' ' )[- 2 ])
1231
+ ) # 2007_000027.xml --> 2007000027
1228
1232
imgs_ann_file_list = [os .path .join (folder_ann , s ) for s in imgs_ann_file_list ]
1229
1233
# logging.info('ANN',imgs_ann_file_list[0::3333], imgs_ann_file_list[-1])
1230
1234
0 commit comments