【问题--Heatmap可视化】B导，请问如何自己手动将注意力热图可视化模块添加DETR模型的预测文件（predict.py）

[Sapphire0904]

毕导您好， 我在参考以往您YOLO系列，其中的注意力热图可视化模块(mode="heatmap")并向DETR模型中手动加入heatmap模块时，遇到如下问题： 请问，我应该如何将 output的shape 由 [batch_size, 100, num_classes + 1]，[batch_size, 100, 4] 转为为YOLO系列（如yolov4-tiny）中的 [b, c, h, w]，如下图所示。