Copy and rename a folder of images into a new folder following the `img_format`. Args: input_folder (str): input folder. output_folder (str): output folder. img_format (str, optional): image format name, do not need extension. Defaults to '%06d'. Ret
(input_folder, output_folder, img_format='%06d')
| 345 | |
| 346 | |
| 347 | def images_to_sorted_images(input_folder, output_folder, img_format='%06d'): |
| 348 | """Copy and rename a folder of images into a new folder following the |
| 349 | `img_format`. |
| 350 | |
| 351 | Args: |
| 352 | input_folder (str): input folder. |
| 353 | output_folder (str): output folder. |
| 354 | img_format (str, optional): image format name, do not need extension. |
| 355 | Defaults to '%06d'. |
| 356 | |
| 357 | Returns: |
| 358 | str: image format of the rename images. |
| 359 | """ |
| 360 | img_format = img_format.rsplit('.', 1)[0] |
| 361 | file_list = [] |
| 362 | os.makedirs(output_folder, exist_ok=True) |
| 363 | pngs = glob.glob(os.path.join(input_folder, '*.png')) |
| 364 | if pngs: |
| 365 | ext = 'png' |
| 366 | file_list.extend(pngs) |
| 367 | jpgs = glob.glob(os.path.join(input_folder, '*.jpg')) |
| 368 | if jpgs: |
| 369 | ext = 'jpg' |
| 370 | file_list.extend(jpgs) |
| 371 | file_list.sort() |
| 372 | for index, file_name in enumerate(file_list): |
| 373 | shutil.copy( |
| 374 | file_name, |
| 375 | os.path.join(output_folder, (img_format + '.%s') % (index, ext))) |
| 376 | return img_format + '.%s' % ext |
| 377 | |
| 378 | |
| 379 | def images_to_array( |
no test coverage detected