-
https://github.com/lyuwenyu/RT-DETR/blob/9f107dab37c402962a4632570e63ec83c30e1ecc/rtdetr_pytorch/src/zoo/rtdetr/rtdetr_criterion.py#L135
在rtdetr_criterion.py的第135行,分类损失被除以num_boxes而不是num_queries:
…
-
Thanks for your star
-
-
1.关于AIFI层的设计
假设现在对极小目标进行检测(大小只占整个图片1%左右),如果去掉AIFI层,对于小目标检测影响大不大?
2.IOU感知查询
Detr存在的问题是object queries是随机的值,导致训练时间长,难以收敛等,IOU感知查询本质是对对象查询初始化的优化处理,消融实验表明IOU感知查询确实有效果,那么还有别的更好都地设计么,如果将AIFI层的输出,选择…
-
您好,请问使用AutoDL服务器训练RT-DETR的话,如何可视化训练损失曲线等等呢???
我安装了visualdl,但是执行visualdl --logdir ./log那个网址打不开
![image](https://github.com/lyuwenyu/RT-DETR/assets/97581575/672a30c1-c4ba-4d71-b47d-62cd1031c823)
-
Hi,
Is it an plan to create a hubconf.py file to make it is easy to use the `torch.hub.load` and use the pre-trained models?
Niknu updated
2 months ago
-
https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetr_pytorch/tools/export_onnx.py Line:78
这个逻辑是不是有点问题,dynamic变成了1的固定batchsize
-
RT-DETR用paddlelite把模型转nb文件时,显示算子都支持,但会报unknown type 0的错误,想问一下还有什么办法在移动端部署吗?
-
yolov5/v8都考虑了一些未包含任何目标的图像作为背景图像的问题。请问[RT-DETR]是否训练时考虑了这个问题?如果是,coco数据集的标注为0吗?
-
Thank you for a great work!
1. Could you please share the DEYO CrowdHuman weights and training config?
2. Have you compared DEYO vs RTDETR performance on the CrowdHuman dataset? I have only notice…