@@ -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