数据集制作代码

修改数据集名称和生成对应的txt文件

#======================================================修改数据集名称和生成对应的txt文件======================================================
file_name = "//home//zy//chair//training-set//middle"     #自建数据集原路径

i = 1

path = "//home//zy//chair//skin_disease_two_classification-master//skin_disease_two_classification-master//dipei_train"     #自建数据集目标路径
for name in os.listdir(file_name):           #遍历原路径下的图片
    sss = "dipei_training" + str(i) + ".png"+",1"+"\n"
    with open("test.txt","a") as f:             #打开并建立一个test.txt文件
        f.write(sss)                            #将信息写入txt文件，个数与图片数量一致

    new_name = name.replace(name, "dipei_training%s.png" % i)     # #修改路径下的图片名称
    os.rename(os.path.join(file_name, name), os.path.join(path, new_name))        #将原来的图片数据通过重命名送到目标路径下.原路径的照片会丢失
    i += 1

print("end")

文字替代

#======================================================文字替代======================================================
with open("test.txt", "r",encoding="utf-8") as f,open("train5.txt", "w", encoding="utf-8") as f2:    #f为原路径，f2为目标路径
    for line in f:     #遍历每行
        if ",malignant" in line:
            line = line.replace(line,"zhongpei_%s.png,malignant"% i+"\n" )
            print("s")
        if ",1" in line:                        #看",1"是否在这一行
            line = line.replace(",1",",malignant")                         #将",1"改成",malignant"
            print("m")
            line1 = line.replace("，1", ",malignant")
        f2.write(line)                          #写入f2
        i=i+1