reademe.txt 272 Bytes
输入图片256*256

预处理:
test_transform = transforms.Compose([
    transforms.Resize((256, 256)),
    transforms.ToTensor(),
    transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]), 
    ])

输出:{'车尾': 0, '车侧': 1, '车头': 2}