🧠AI 冷知识
你知道吗?
这些有趣的 AI 小知识可能让你大吃一惊
👁️
预训练权重的力量torchvision.models里的预训练模型在ImageNet的128万张图片上训练过,蕴含了极其丰富的视觉特征。用它们做迁移学习,有时只需几十张目标图片就能达到从零训练几万张才能达到的效果——这就是"站在巨人肩膀上"的真正含义。
一句话总结
💡
torchvision 提供常用视觉数据集、预训练模型与图像变换工具,是计算机视觉开发的标配库。
常见误区
这些坑别踩
✗
误区 1
torchvision 只能用于图像分类。
✓
正确理解
它还提供检测、分割模型及变换工具,覆盖多种视觉任务。
✗
误区 2
预训练模型的输入可以是任意尺寸图片。
✓
正确理解
多数预训练模型要求固定尺寸(如 224x224)并做相同归一化,否则结果异常。
✗
误区 3
transforms.ToTensor() 会做归一化到 0-1 以外。
✓
正确理解
它仅将 HWC 转为 CHW 并缩放到 [0,1],归一化需额外 Normalize。