Skip to content

Commit 9ce9ae4

Browse files
committed
fix mask naming to prevent name overlap, and potential overwrites
1 parent 47d2f2c commit 9ce9ae4

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

code/preprocess_ims.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -61,13 +61,15 @@ def move_labels(input_folder, output_folder, fold_id):
6161

6262
for e,i in enumerate(cam2):
6363
fname_parts = i.split(os.sep)
64-
cam2_base = str(fold_id) + '_' + fname_parts[-1]
64+
65+
# Thanks @Nitish for these fixes:)
66+
cam2_base = str(fold_id) + '_' + fname_parts[-3] +'_' + fname_parts[-1]
6567

6668
fname_parts = cam3[e].split(os.sep)
67-
cam3_base = str(fold_id) + '_' + fname_parts[-1]
69+
cam3_base = str(fold_id) + '_' + fname_parts[-3] +'_' + fname_parts[-1]
6870

6971
fname_parts = cam4[e].split(os.sep)
70-
cam4_base = str(fold_id) + '_' + fname_parts[-1]
72+
cam4_base = str(fold_id) + '_' + fname_parts[-3] +'_' + fname_parts[-1]
7173

7274
shutil.copy(i, os.path.join(output_folder,cam2_base))
7375
shutil.copy(cam3[e], os.path.join(output_folder,cam3_base))
@@ -84,7 +86,7 @@ def move_png_to_jpeg(input_folder, output_folder, fold_id):
8486
for i in cam1_files:
8587
cam1 = misc.imread(i)
8688
fname_parts = i.split(os.sep)
87-
cam1_base = str(fold_id) + '_' +fname_parts[-3]+fname_parts[-1].split('.')[0] + '.jpeg'
89+
cam1_base = str(fold_id) + '_' +fname_parts[-3] + '_' + fname_parts[-1].split('.')[0] + '.jpeg'
8890
misc.imsave(os.path.join(output_folder, cam1_base), cam1, format='jpeg')
8991

9092

0 commit comments

Comments
 (0)