pytorch 迁移学习实战 天气识别
2023-09-27 14:25:47 时间
代码:pytorch迁移学习实战,天气识别-深度学习文档类资源-CSDN文库
dataset
import random
import numpy as np
import torch
from torchvision import transforms
import torchvision
if __name__=='__main__':
import matplotlib.pyplot as plt
# 定义数据处理transform
transform = transforms.Compose([
transforms.Resize([224,224]),
transforms.ToTensor()
])
#transform是对特征进行变换,target_transform是对标签进行变换
dataset=torchvision.datasets.ImageFolder(root='./data',transform=transform,
target_transform=torchvision.transforms.Lambda(lambda x:torch.zeros(4,dtype=torch.float).scatter_(0,torch.tensor(x),value=1)))
print(dataset.classes)
print(dataset.class_to_idx)
print(dataset.imgs
相关文章
- 【FPGA教程案例100】深度学习1——基于CNN卷积神经网络的手写数字识别纯Verilog实现,使用mnist手写数字数据库
- 【车牌识别】基于HOG特征提取和GRNN网络的车牌识别算法matlab仿真
- 【CMAC】基于CMAC小脑模型的人体行走姿态识别算法仿真
- 基于matlab车牌识别算法
- 在 Linux 系统里识别 USB 设备名字的 4 种方法
- OCR识别服务开发手册
- NetworkMiner识别原理
- 6、Halcon图像边缘提取和轮廓识别
- mybatis参数为0识别为空字符串的查询处理
- 完整opencv(emgucv)人脸、检测、采集、识别、匹配、对比
- ZXing.Net.Mobile无法识别较大的条码
- 《ANTLR 4权威指南》——1.2 运行ANTLR并测试识别程序
- 将tensorflow训练好的模型移植到Android (MNIST手写数字识别)
- ORA-12521: TNS: 监听程序当前无法识别连接描述符中请求的实例(原)
- QTP不识别树结构中的点击事件
- Vuforia 制作AR识别类项目过程简介