神经网络黑匣子(Mobilenetv3)——中间层的提取和通过CV2显示可视化中间层代码

导入神经网络模型 

 导入神经网络模型函数

1,通过循环 特征tensor x 依次通过大整个模型中的各个模块按照名字进行中间层提取 存放到列表中。

y=x[1:,:,:,:]
d = y.permute(0, 2, 3, 1).squeeze(0)

# cv2.imshow("y",y)

y=d.numpy()            # 这里是转ndarray的一个方法
cv2.imshow("Crop", y)  # str(name)需要进行一个字符串的转换
cv2.waitKey(0)
cv2.destroyAllWindows()

 

将提取的中间特征tensor 转换为ndarray 然后进行cv2.imshow 就可以了参照上面的代码