请问灰度图像输入VGG，怎么改才是最好的？

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

Railgun 发布于2021-01

如果不改会对结果有影响吗？

全部评论(27)

thinc

#22 回复于2021-01

AIStudio810259 #13

可以搞，具体做法就是直接将第一个卷积层的输入通道改为1

你这个是自己组的网，用的api是nn.Conv2D()

他这里用的官方搭好的网络，api是vision.models.vgg16()

thinc

#23 回复于2021-01

AIStudio810258 #16

预训练模型能只用中间几层么？

vision.models.vgg16()这种预训练模型都是论文中的模型，也就是从ImageNet上训练出来的

thinc

#24 回复于2021-01

AIStudio810258 #17

既然输出维度可以改，那么同样的方法改输入维度呢？。只用预训练模型的中间层，第一层和最后一层都从新训。当然，改第一层和改最后一层的意义大大不同。

这样的话我觉得你得把api的源代码修改一下，然后重新调用了= =

AIStudio810258

#25 回复于2021-01

thinc #24

这样的话我觉得你得把api的源代码修改一下，然后重新调用了= =

如果只是预训练模型呢？可以只用中间层么～～

thinc

#26 回复于2021-01

AIStudio810258 #25

如果只是预训练模型呢？可以只用中间层么～～

不行= =

AIStudio810258

#27 回复于2021-01

thinc #26

不行= =

最近看到了一个文章以前做过的消融实验，研究迁移学习的。

挨个交换模型的每一层，看看那些层能迁移。。。

AIStudio810258

#28 回复于2021-01

thinc #26

不行= =

迁移学习好玩么？