【2】人工智能训练 - 使用模型及保存预测结果

import torch from torchvision import models import csv from sklearn.metrics import classification_report import cv2 as cv import os device = torch.dev