关于删除检测头的问题

[newnoobboy]

B导，您好，我想问一下如果想在YOLOV5模型中删除一个尺度的检测头，我除了修改anchors_mask，anchors之外，还要在模型的那块去修改代码呢？

[bubbliiiing]

是的，要修改模型的代码，减少一个预测头

[newnoobboy]

我在yolo.py文件中把P5的检测头删除了，然后在dataloader.py文件里的get_target方法中修改了input_shape // {0:16, 1:8,}之后就一直检测不到目标了，是还需要在别的文件下修改呢，还是我修改的部分是错误的呢

[bubbliiiing]

不一般都删除最上面那个检测头吗……

[newnoobboy]

导，您好，我有点不理解这个代码input_shape // {0:32, 1:16, 2:8, 3:4}[l],如果是原模型的三个检测头不是只用到了0，1，2这三个序列嘛？我用两个检测头的话，不是应该只用到了0，1两个序列，为什么如果我改成input_shape // {0:16, 1:8}[l]这样的话就检测不到目标呢，但是如果我改成input_shape // {0:16, 1:8, 2:8, 3:4}[l]这样的话就可以检测到目标呢，真心求解答

[bubbliiiing]

这我也不太清楚，我可能需要看看代码

[BILIBILINB]

B导，您好，我想问一下在YOLOV8模型中删除小目标的检测头，但是最后模型的参数增加了，大小也增加了，为什么呢？