eac85cd5
 Hu Chunming
调通va
|
1
2
3
4
5
6
7
8
9
10
|
输入图片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}
|
