根据得到的分类结果数据，对于给出的测试图像进行标签重命名：
测试集分类结果文件：stag_01_test.csv
测试图片文件夹：stag_01_test
##########图片重命名 Rename，设置标签########## import os ROOT_DIR = os.path.abspath("./") img_path = os.path.join(ROOT_DIR, "stag_01_test") imglist = os.listdir(img_path) print('img_path',img_path) #########对于csv文件进行读取############### import csv t1 = open('stag_01_test.csv', 'r') fileone = t1.readlines() t1.close() print(fileone[1]) print(fileone[1].strip().split(',')) print(fileone[1].strip().split(',')[0]) print(fileone[1].strip().split(',')[1]) ###########标签解析############## def result_(res): if res=='airplane': return 0 elif res=='ship': return 1 elif res=='bridge': return 2 elif res=='oilcan': return 3 elif res=='build': return 4 else: return 5 i = 0 for img in imglist: if img.endswith('.jpg'): # print(i) src = os.path.join(os.path.abspath(img_path), img) # original name print('src',src) #src /home/jovyan/work/data/gyx/Test_all/stag_01_test/0001.jpg img_n = img[:-4] print(img_n) #0001 filename=fileone[i+1].strip().split(',')#['0001.jpg', 'airplane'] if filename[0]==img: id = int(result_(filename[1])) else: print('Wrong filename') dst = os.path.join(os.path.abspath(img_path),img_n + '_%d'%(id) + '.jpg') # original name + _1 os.rename(src, dst) # rename==>recover the original name i+=1原始图像文件：
加上标签的图像文件：
