可结合着一下两个看一下:

Pytorch:利用预训练好的VGG16网络提取图片特征

pytorch实现用VGG对图片的特征提取及可视化

对于自己训练好的网络,在参考以上两篇文章的基础上要根据自己的情况调整,

耐心去理解下原理就明白了。

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐