API_Net代码的train.txt与val.txt

import os

if __name__=="__main__":
    path = "/home/yjys/datasets/CUB_200_2011"
    train_test_file = "train_test_split.txt"
    image_class_label = "image_class_labels.txt"
    image = "images.txt"
    image_path = "/home/yjys/datasets/CUB_200_2011/images"

    image_name = []
    with open(os.path.join(path,image), 'r') as f:
        for line in f.readlines():
            img_name = os.path.join(image_path, line.strip("\n").split(" ")[1])
            image_name.append(img_name)

    train_test = []
    with open(os.path.join(path, train_test_file)) as f:
        for line in f.readlines():
            t_t = line.strip("\n").split(" ")[1]
            train_test.append(int(t_t))

    img_class_label = []
    with open(os.path.join(path, image_class_label)) as f:
        for line in f.readlines():
            c_l = line.strip("\n").split(" ")[1]
            img_class_label.append(int(c_l))

    with open("train.txt", 'w') as f:
        with open("val.txt", 'w') as fv:
            for index, name in enumerate(image_name):
                if train_test[index]:
                    f.write(name+" "+str(img_class_label[index]-1)+"\n")
                else:
                    fv.write(name+" "+str(img_class_label[index]-1)+"\n")


#IndexError: Target 200 is out of bounds.使用cpu炮程序后，出现的错误，也就是在计算🔠input和targets的损失函数的时候出现的错误
#RuntimeError: cuda runtime error (710) : device-side assert triggered 使用gpu炮程序，出现的错误，看不出来