detr-pytorch

detr-pytorch copied to clipboard

运行报错

5

b导，我在跑您的detr代码时，训练10个epoch没问题，但是在每十个epoch计算一次mAP的时候报错： File "/opt/data/private/detr/utils/utils_bbox.py", line 31, in forward boxes = boxes * scale_fct[:, None, :] RuntimeError: expected device cuda:0 and dtype Float but got device cuda:0 and dtype Long 这是怎么回事呢？

ChristmasLee

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

1

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

Sapphire0904

想问下导，针对detr的话，喂入网络的图像size应该是设置多少比较合适呢，代码里给的是800*800，针对detr设置的size大小有没有明确要求呢，比如16的倍数，我目前手头的数据原尺寸是1920*1080，能否给些resize大小的建议，谢谢

3

Shawn2020-cmd

训练到第10个epoch时出错！

7

 在训练到第10个epoch时，进行mAP评估出现上述问题。

ljh94o

训练VOC数据集中的4个类，报错

15

 B导过年好，我抽取了VOC数据集中的4个类，用这个数据集训练过其他检测器都没有问题，但您DETR报了这个错，请问是什么问题呀

1e12Leon

VOC2007数据集上跑了50多个epoch，Total Loss和Val Loss数值一直保持在4点多不下降

lumia-cracker

RuntimeError

1

RuntimeError: Error(s) in loading state_dict for DETR: size mismatch for class_embed.weight: copying a param with shape torch.Size([6, 256]) from checkpoint, the shape in current model is torch.Size([92, 256]). size mismatch...

WangYafei0904