Caltech-UCSD Birds-200-2011資料集類別資訊轉換成列表和字典形式

# 列表型別
f = open('classes.txt', 'r')
re = []
for i in f.readlines():
    lst = i.strip().split(".")
    if(lst[1] not in re):
        re.append(lst[1])
f.close()
# print(re)

# fromfile.txt來源
# for index, item in enumerate(re):
    # print(index, item)

# 字典型別
dt = {}
with open('fromfile.txt', 'r') as dict_file:
    for line in dict_file:
        (key, value) = line.strip().split(' ')
        dt[key] = value
# with open語句不必呼叫f.close()方法
print(dt)

# 列表寫入txt檔案
# f1 = open('new_classes.txt', 'w')
# for s in re:
#     f1.write(s)
#     f1.write('\n')
# f1.close()
 

classes.txt是CUB2011資料集自帶的類別資訊檔案

fromfile.txt來源split_classes.py文中的兩行程式碼

生成列表型別和字典型別也在split_classes.py中