18143453325 在线咨询 在线咨询
18143453325 在线咨询
所在位置: 首页 > 营销资讯 > 行业动态 > 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类的分类器模型,但是这些分类与我们所使用的分道物基本没有关联,因此以这个使用这个模型为基础的迁移学习并没有明显的帮助。

如果想将前面的避撞功能加到这个循路应用里,有两种方法可以实现:

  1. 将避撞项目收集好的 blocked 与 free 类别图像数据,分别加到这个项目的对应目录中一起进行模型训练。
  2. 运用迁移学习技巧,在执行模型训练之前将避障项目的best_model.pth导进来作为训练的基础,当然您得先把避撞的best_model.pth模型文件复制到这个目录下。不过这个做法需要在代码上做些小幅度的修改,将

    关键词:

74
73
25
news

版权所有© 亿企邦 1997-2022 保留一切法律许可权利。

为了最佳展示效果,本站不支持IE9及以下版本的浏览器,建议您使用谷歌Chrome浏览器。 点击下载Chrome浏览器
关闭