-
Notifications
You must be signed in to change notification settings - Fork 1.1k
[Feat]: Add dynamic graph model RT-DETR-L #4801
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: develop
Are you sure you want to change the base?
Conversation
Bobholamovic
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
- 需确认是否使用了pre-commit进行自动代码风格检查;
- 新增代码量超过了9000行,需评估这些代码是否都是必要的;
- 需评估RT-DETR-L是否应该只是RT-DETR的一个变体(通过配置文件确定),考虑到transformers库是这样设计的(https://huggingface.co/docs/transformers/main/model_doc/rt_detr)。
paddlex/inference/models/object_detection/modeling/RT_DETR_L.py
Outdated
Show resolved
Hide resolved
paddlex/inference/models/object_detection/modeling/RT_DETR_L.py
Outdated
Show resolved
Hide resolved
paddlex/inference/models/object_detection/modeling/RT_DETR_L.py
Outdated
Show resolved
Hide resolved
paddlex/inference/models/object_detection/modeling/RT_DETR_L.py
Outdated
Show resolved
Hide resolved
paddlex/inference/models/object_detection/modeling/RT_DETR_L.py
Outdated
Show resolved
Hide resolved
| __all__ = ["HungarianMatcher"] | ||
|
|
||
|
|
||
| class GIoULoss(object): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
目前代码量有点儿太多了,我们只是做推理的话,loss部分是否可以不迁移进来?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
RT-DETR-L模型组件HungarianMatcher依赖GIoULoss运行,此处需要进行迁移
| return out_bbox if format == "xyxy" else bbox_xyxy_to_cxcywh(out_bbox) | ||
|
|
||
|
|
||
| def varifocal_loss_with_logits( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这里也有不少loss相关内容
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
RT-DETR-L模型组件DINOHead、RTDETRTransformer组件依赖utils中各loss相关方法运行,此处需要进行迁移
No description provided.