Imageneイメージフォルダをカテゴリ番号で命名する

929 ワード

ImageNetは偉大な画像認識データベースです
しかし、その中の各カテゴリの画像のフォルダ名は強迫症患者に耐えられない.
カテゴリフォルダ名を番号に変更するウィジェットをコードしました
 
###
imagenet_class_index.jsonはImagenet画像のカテゴリ表記jsonファイルです
Imagenetフォルダに解凍した1000個のフォルダを置く
import json
import os

#     json  
file=open("imagenet_class_index.json",'r')
load_dict = json.load(file)

print(load_dict)

#         
class_dict={}
for item in load_dict:
    #         ,        ,      
    class_dict[load_dict[item][0]]=item

#  Imagenet     
ImageNet_Path='Imagenet'
dir_list=os.listdir(ImageNet_Path)

for dir_name in dir_list:
    #                (      )
    class_index_in_str=class_dict.get(dir_name)
    if class_index_in_str is None:
        continue
    else:
        #  linux        
        command = 'mv '+os.path.join(ImageNet_Path,dir_name)+' '+os.path.join(ImageNet_Path,class_index_in_str)
        os.system(command)