摘要: Pytorch常用代码整理 查看Pytorch基本信息 需要用到的包 import collections import os import shutil import tqdm import numpy as np import PIL.Image import torch import torc 阅读全文
posted @ 2021-10-20 13:56 梁君牧 阅读(285) 评论(0) 推荐(0) 编辑
摘要: Pytorh数据集加载框架 图像分类的数据集加载 以加载mnist手写数字数据集为例。 首先将官网的mnist格式通过以下代码存储为图片。 # -*- coding: UTF-8 -*- # 把mnist数据集转成图片做测试,图片更为通用 import cv2 import os from kera 阅读全文
posted @ 2021-10-20 10:44 梁君牧 阅读(148) 评论(0) 推荐(0) 编辑