pytorch_基于CIFAR创建自己的数据集并显示图像
摘要:
创建完自己的数据集训练完之后,开始了预测,如何看到预测图像 Next import cv2 用来显示图像 在原有测试数据文件中 增加图像显示功能 def TestingData(self): model_net = torch.load('model_shanbu_128.pkl') self.Lo 阅读全文
posted @ 2020-05-29 14:40 wangxiaobei2019 阅读(444) 评论(0) 推荐(0) 编辑