| 50 | |
| 51 | |
| 52 | def read_filepaths(args): |
| 53 | dirnames = sorted(os.listdir(args.root_input_directory)) |
| 54 | logger.info(f"Total {len(dirnames)} dirs: {dirnames}") |
| 55 | all_filepaths = [] |
| 56 | for dirname in dirnames: |
| 57 | dirpath = os.path.join(args.root_input_directory, dirname) |
| 58 | if not os.path.isdir(dirpath): |
| 59 | continue |
| 60 | logger.info(f"------------------------- class {dirname}") |
| 61 | filenames = sorted(os.listdir(dirpath)) |
| 62 | logger.info(f"------------------------- Total {len(filenames)} files") |
| 63 | |
| 64 | for filename in filenames: |
| 65 | filepath = os.path.join(dirpath, filename) |
| 66 | all_filepaths.append(filepath) |
| 67 | return all_filepaths |
| 68 | |
| 69 | |
| 70 | def main(args): |