1 前言

在此次的自动驾驶任务中,我们需要完成图像识别的任务;

2 模型选择——

3 模型修改——修改最后一层FC层

具体的写作方式可以参考博文——《Pytorch学习笔记(I)——预训练模型(一):加载与使用》

更多推荐