yolov3代码疑问

Question

yolov3的代码中

self.register_buffer('anchor_grid', a.clone().view(self.nl, 1, -1, 1, 1, 2)) 

这一行是什么意思，在做什么

Answer 1

这个和pytorch中的register_buffer有关，相当于缓存一些参数，这样可以将这些原本非模型本身的参数缓存到模型当中，更为具体的解释和代码注释你可以看下下面这个，虽然是yolov5的，但是这部分内容基本上是一样的

yolov5 代码问题请教 · Issue #13 · DefTruth/lite.ai.toolkit · GitHub yolov5是一个基于anchor的算法，再前向推理的过程中应该涉及anchor的计算，但是在代码中没有看到andhor的任何信息，请问是怎么处理的？ https://github.com/DefTruth/lite.ai.toolkit/issues/13