基于Resnet50的pytorch框架下的图像特征提取

在Resnet50CNN结构下实现图像的特征提取,这里采用的是CV2的图像读入方式,最后再把获得的图像转换成npy格式进行输出得得,图像对应的特征。网络 # -*- coding: utf-8 -*- """ Function: 图像特征的提取,能够依据需求修改CNN的输出,获得不一样层网络的输出图像特征 Writer: Zenght date:2019.2.16 """ from __futu
相关文章
相关标签/搜索