实用工具
文章平均质量分 86
CV51
对计算机视觉中的目标检测、跟踪、分割、轻量化神经网络感兴趣
展开
-
5分钟搞定深度学习中间特征可视化
这里直接使用torchvision中带有预训练权重的resnet18,可视化resnet18某一层的输出feature map。import cv2# 从torchvision中导入resnet16。原创 2023-10-31 11:45:29 · 369 阅读 · 0 评论 -
如何优雅地读取网络的中间特征?
在调试深度神经网络工程时,常会在前向计算过程中将网络的中间层信息返回,便于打印或者可视化网络中间结果。实现该功能的一个常用方法是在构建model类时,在forward返回要保留的中间信息。这里跟大家分享一个更优雅、便捷的方法,利用torchvision提供的类,在网络前向计算时返回指定的特征。原创 2023-10-16 11:20:35 · 173 阅读 · 0 评论