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