中国交通标志识别 德国交通标志识别

import torchimport cv2import torchvision.transforms as transformsimport torch.nn as nnimport torch.optim as optimimport argparsefrom load_data import Traffic# 定义是否使用GPUdevice = torch.device("cuda" if torch