pic should be PIL Image or ndarray, Got numpy.ndarray:CSV 表格别走图像 ToTensor

简介: 本文剖析PyTorch中误用图像ToTensor处理表格数据的典型错误:一维标签触发类型检查异常,整表append导致样本数为1。强调应按行取样,用torch.from_numpy直接转数值张量,避免强行适配图像API,并验证数据粒度、形状与模型语义一致性。(239字)

原帖读取 CSV 后,把整张特征表和一维标签分别 append 到列表,再对两者调用图像 ToTensor:

TypeError: pic should be PIL Image or ndarray. Got <class 'numpy.ndarray'>

这题有两处数据集设计问题

一维类别数组被交给图像变换,触发旧 ToTensor 的维数检查;同时整张表 append 为一个列表元素,使样本数变成 1。只把标签升成二维图片或换一个 ToTensor 版本,仍保留了错误的数据粒度。先明确每行是一条样本,再分别处理特征与目标。

错误虽显示 numpy.ndarray,但旧版 ToTensor 还会检查数组维数;一维标签不是它期望的图像。另一个问题是 append 整表后 len(self.features) 只有 1,数据集不再按行取样。

先在数据集外检查原始表格:

import pandas as pd
frame = pd.read_csv("android.csv")
features = frame.iloc[:, :-1]
labels = frame.iloc[:, -1]
print("josh050608")
print("features:", features.shape, features.dtypes.tolist())
print("labels:", labels.shape, labels.dtype, labels.head().tolist())

这类非图像数据直接转换数值数组,不使用 torchvision.transforms.ToTensor。若末列是已确认的整数类别编号,可以改成:

import torch
from torch.utils.data import TensorDataset, DataLoader
x = torch.from_numpy(features.to_numpy(dtype="float32"))
y = torch.from_numpy(labels.to_numpy(dtype="int64"))
dataset = TensorDataset(x, y)
train_loader = DataLoader(dataset, batch_size=32, shuffle=True)

若标签是字符串,先建立稳定的类别映射;若是回归目标,保留浮点类型并配合回归损失。不要为了让转换成功而把缺失值和脏字符串默默变成 0。

原帖还计划用 Conv1d:表格张量 [N,F] 不会自动具有卷积所需的 [N,C,L] 语义,需按模型定义决定是否加通道维,不能任意 reshape。验证 len(dataset) 等于 CSV 行数、单个样本为一行特征,再检查第一个 batch 的形状与标签配对。

针对旧 torchvision 0.2.2,应核对该版本源码中的图像维数检查;现代图像变换接口的支持范围不能反推旧版行为。

直接用数值表格验证取样粒度

下面用合成 NumPy 数组演示四条样本、每条三个特征。它不读 android.csv,也不依赖 torchvision:

import numpy as np
import torch
from torch.utils.data import TensorDataset, DataLoader
feature_array = np.arange(12, dtype=np.float32).reshape(4, 3)
label_array = np.array([0, 1, 0, 1], dtype=np.int64)
dataset = TensorDataset(torch.from_numpy(feature_array), torch.from_numpy(label_array))
assert len(dataset) == 4
assert dataset[0][0].shape == (3,)
assert dataset[0][1].ndim == 0
loader = DataLoader(dataset, batch_size=2, shuffle=False)
x, y = next(iter(loader))
assert x.shape == (2, 3) and y.shape == (2,)
assert torch.equal(y, torch.tensor([0, 1]))
print("sample/batch:", dataset[0][0].shape, x.shape, y.shape)

真实 CSV 先检查缺失值、字符串列与目标类型,分类编号转 int64,连续回归目标则保留适合损失的浮点格式。原帖的 features.append(整表) 应去掉,或者让自定义 Dataset 直接保存二维矩阵并用行索引返回。

若后续模型是 Conv1d,需要为 [N,F] 中的 F 指定序列语义。作为单通道长度 F 的输入可以明确 x.unsqueeze(1) 成 [N,1,F];若每列并非有序信号,应先判断卷积是否适合,不能仅为满足 API 拼维度。

本文表格转换示例按所引用 pandas 版本接口写;维护旧 pandas 时使用该版本支持的数组获取方式。验收数据集长度、单条 shape、一个 batch 的样本与标签顺序,再做模型前向,避免旧图像变换异常消失后仍在用“一条整表样本”训练。

参考:原始求助帖;技术依据:PyTorch 数据加载(2.5)、torch.from_numpy(2.5)、torchvision ToTensor(0.20)、torchvision 0.2.2 变换源码、pandas 2.2 to_numpy。

相关文章
|
10天前
|
Linux 开发工具 iOS开发
Codex保姆级教程(下载+安装+登录+使用,含所有版本,2026最新)
Codex是OpenAI推出的开源AI编程智能体,本地运行,可理解整个代码库,支持自然语言指令改代码、执行命令、跑测试、修Bug,兼容Win/macOS/Linux。(239字)
4057 20
|
11天前
|
JavaScript API iOS开发
DeepSeek Harness下载、安装、配置一篇搞定(2026整理,送常用插件)
DeepSeek Harness(DSH)是DeepSeek于2026年开源的MIT协议AI Agent框架,支持插件化组装模型、工具、沙箱与UI,兼容主流大模型。本文提供免Node.js的桌面端安装包(Win/macOS),含详细图文教程及热门插件推荐。
4158 49
|
6天前
|
存储 弹性计算 人工智能
2026 年阿里云双十一活动入口攻略,新客服务器优惠通道、三大会场入口梳理
综合来看,2026年阿里云双十一活动入口清晰: 阿里云轻量应用服务器、阿里云ECS云服务器、阿里云GPU云服务器、阿里云百炼大模型服务平台 活动会场、综合权益中心、99计划专场三大入口覆盖服务器、AI权益与续费特价;新用户通过控制台首页banner、权益中心、产品页三处路径进入专属专场,老用户通过控制台弹窗和费用中心权益中心进入老客专区,企业用户做企业实名后解锁专属满减与阶梯折扣。
820 0
|
11天前
|
iOS开发 Docker MacOS
【2026实测】Docker Desktop下载、安装、汉化、配置教程(附官网安装包)
Docker Desktop 是 Docker 官方推出的桌面容器管理工具,集成引擎、CLI、图形界面与 Kubernetes,支持 Win/macOS/Linux。2026 最新版 4.88.1 强化 ARM 原生运行、镜像加速及资源可视化,附中文汉化与国内镜像配置指南。
1996 7
|
6天前
|
云安全 人工智能 安全
|
5天前
|
Linux 开发工具 iOS开发
2026最新codex下载安装保姆级使用教程(下载+安装+登录+使用,含所有历史版本)
本文详解OpenAI Codex国内快速上手指南:提供Win/macOS/Linux全平台免安装包、npm/Homebrew等多方式安装、账号/API Key双登录、简体中文切换,覆盖下载、安装、汉化、使用及高频避坑(如微软商店失败、npm超时等),10分钟即可跑通。
1112 3
|
12天前
|
安全 算法 定位技术
《高性价比人生指南》完整原版:普通人用最少的资源,过好这一生的底层逻辑~How To Live Better
📖《高性价比人生指南》是一份基于实证、拒绝焦虑的生存手册,倡导以“人生时薪”为视角,重新计算时间、金钱与精力的成本与收益。它不承诺逆袭,只教人算清账:省下的钱若耗掉时间便亏;买体验比买物品更值;睡眠是最高杠杆的健康投资;关系贵在情绪价值而非表面热络;学习重在改变行为而非自我安慰。核心理念:用低成本、高回报的微小行动,换取长期的选择权——不是成为谁,而是掌控自己人生的主动权。
2145 6
|
云安全 安全 网络安全
|
3天前
|
人工智能 安全 API
DeepSeek Harness下载、安装、配置一篇搞定(2026整理,送常用插件)
DeepSeek Harness(dsh)是DeepSeek AI于2026年开源的轻量级Agent运行框架,基于Cordis内核,践行“一切皆插件”理念。支持Web/桌面双端,开箱即用;内置多模型接入、丰富插件生态(超3.5万),适配编程、办公、创作等场景。(239字)
707 0

热门文章

最新文章