2020年5月29日

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 阅读(442) 评论(0) 推荐(0) 编辑

pytorch_训练CIFAR数据集

摘要: cifar数据集训练与测试 CIFAR数据集下载 https://download.csdn.net/download/wangxiaobei2017/12474160 # !/usr/bin/env python # -*- coding: utf-8 -*- import torch impor 阅读全文

posted @ 2020-05-29 14:34 wangxiaobei2019 阅读(994) 评论(0) 推荐(1) 编辑

pytorch_基于cifar创建自己的数据集并训练

摘要: 基于cifar创建自己的数据集并训练 整体流程 resize图像,在图像名称附上标签 eg: 1_a.jpg 获取图像的像素 R G B 合并在一个列表 将文件名、数据、标签和batch_label合并在一个字典中 用二进制的方式存储字典内容,得到和cifar相同的 修改 batches.meta 阅读全文

posted @ 2020-05-29 14:17 wangxiaobei2019 阅读(852) 评论(0) 推荐(0) 编辑

导航