摘要: yolov5内存分布分析 Transpose输出分析 假设batch_size为1,yolov5有三个输出,shape分别是: (1,3,80,80,85) (1,3,40,40,85) (1,3,20,20,85) 其中3代表anchor数量,20*20代表feature_map大小,85代表bo 阅读全文
posted @ 2024-06-12 13:50 eastgeneral 阅读(7) 评论(0) 推荐(0) 编辑