所在位置:
首页 >
营销资讯 >
行业动态 > model = models.alexnet(pretrained=True)
model = models.alexnet(pretrained=True)
时间:2022-03-16 15:48:01 | 来源:行业动态
时间:2022-03-16 15:48:01 来源:行业动态
用到PyTorch预训练好的AlexNet 1000类的分类器模型,但是这些分类与我们所使用的分道物基本没有关联,因此以这个使用这个模型为基础的迁移学习并没有明显的帮助。
如果想将前面的避撞功能加到这个循路应用里,有两种方法可以实现:
- 将避撞项目收集好的 blocked 与 free 类别图像数据,分别加到这个项目的对应目录中一起进行模型训练。
- 运用迁移学习技巧,在执行模型训练之前将避障项目的best_model.pth导进来作为训练的基础,当然您得先把避撞的best_model.pth模型文件复制到这个目录下。不过这个做法需要在代码上做些小幅度的修改,将