Pytorch Kaggle CSV文件

Pytorch Kaggle CSV文件,pytorch,kaggle,Pytorch,Kaggle,我已经在kaggle图像数据集上训练了一个预先训练过的Resnet50分类器,并将其标记为0表示无新冠病毒,1表示胸部疾病,2表示新冠病毒。 现在为了比赛,我需要将这些标签保存在CSV文件中。 当我运行以下代码来评估测试图像时 #从新冠病毒文件夹中读取测试图像并进行预测 总数=0 新冠病毒阳性=0 对于os.walk(“/content/gdrive/My Drive/dlai3-phase3/VALIDATE/VALIDATE”,top-down=False)中的根目录、目录和文件: 对于文件

我已经在kaggle图像数据集上训练了一个预先训练过的Resnet50分类器,并将其标记为0表示无新冠病毒,1表示胸部疾病,2表示新冠病毒。 现在为了比赛,我需要将这些标签保存在CSV文件中。 当我运行以下代码来评估测试图像时

#从新冠病毒文件夹中读取测试图像并进行预测
总数=0
新冠病毒阳性=0
对于os.walk(“/content/gdrive/My Drive/dlai3-phase3/VALIDATE/VALIDATE”,top-down=False)中的根目录、目录和文件:
对于文件中的名称:
总imgs+=1
#im_covid=Image.open(“数据/测试/covid/03BF7561-A9BA-4C3C-B8A0-D3E585F73C.jpeg”)
img=Image.open(os.path.join(root,name))
img=img.convert('RGB')
pred,probs=预测新冠病毒(img)
打印(pred)
打印(“新冠病毒阳性测试集的准确性:{}”。格式(新冠病毒阳性/总imgs))
但它打印的是字符串标签,而不是数字

有人能告诉我怎样才能得到数字标签吗