《Pytorch模型推理及多任务通用范式》_第三节课
1 课程学习
本节课主要对于大白AI课程https://mp.weixin.qq.com/s/STbdSoI7xLeHrNyLlw9GOg
《Pytorch模型推理及范式》课程中第三节课进行学习
2 作业解答
必做题
把模型改为 resnet18,加载相应的模型权重(Lesson2 的物料包中有),跑一下 0.jpg
和 1.jpg,看一下输出结果。官方 torchvision 训练 mobilenet 和训练 resnet 的方式是一样
的,所以数据预处理和数据后处理部分完全相同
(‘umbrella’, 0.9998866319656372) (‘peacock’, 0.9999724626541138)
自己找 2 张其他图,用 resnet18 做下推理。 (‘Persian cat’, 0.2359200119972229) (‘airliner’, 0.615960538387
共有 0 条评论