知了AI学习平台Logo知了
🧠AI 冷知识

你知道吗?

这些有趣的 AI 小知识可能让你大吃一惊

👁️
预训练权重的力量

torchvision.models里的预训练模型在ImageNet的128万张图片上训练过,蕴含了极其丰富的视觉特征。用它们做迁移学习,有时只需几十张目标图片就能达到从零训练几万张才能达到的效果——这就是"站在巨人肩膀上"的真正含义。

一句话总结
💡

torchvision 提供常用视觉数据集、预训练模型与图像变换工具,是计算机视觉开发的标配库。

常见误区

这些坑别踩

✗

误区 1

torchvision 只能用于图像分类。

✓

正确理解

它还提供检测、分割模型及变换工具,覆盖多种视觉任务。

✗

误区 2

预训练模型的输入可以是任意尺寸图片。

✓

正确理解

多数预训练模型要求固定尺寸(如 224x224)并做相同归一化,否则结果异常。

✗

误区 3

transforms.ToTensor() 会做归一化到 0-1 以外。

✓

正确理解

它仅将 HWC 转为 CHW 并缩放到 [0,1],归一化需额外 Normalize。