(path)
| 91 | |
| 92 | |
| 93 | def get_image_list(path): |
| 94 | image_names = [] |
| 95 | for maindir, subdir, file_name_list in os.walk(path): |
| 96 | for filename in file_name_list: |
| 97 | apath = osp.join(maindir, filename) |
| 98 | ext = osp.splitext(apath)[1] |
| 99 | if ext in IMAGE_EXT: |
| 100 | image_names.append(apath) |
| 101 | return image_names |
| 102 | |
| 103 | |
| 104 | def write_results(filename, results): |