- 博客(2)
- 收藏
- 关注
原创 YOLO后处理
首发于个人博客理论分析system.pngYOLO从v2版本开始重新启用anchor box,YOLOv2网络的网络输出为尺寸为[b,125,13,13]的tensor,要将这个Tensor变为最终的输出结果,还需要以下的处理:解码:从Tensor中解析出所有框的位置信息和类别信息NMS:筛选最能表现物品的识别框解码过程解...
2018-10-23 13:13:00 5091
原创 神经网络压缩实验-Deep-compression
首发于个人博客,结合论文阅读笔记更佳实验准备基础网络搭建为了实现神经网络的deep compression,首先要训练一个深度神经网络,为了方便实现,这里实现一个两层卷积+两层MLP的神经网络class net(pt.nn.Module): def __init__(self): super(net,self)...
2018-10-05 17:22:00 1659 8
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人