Update: support DCNv4 in InternImage!

OpenGVLab / InternImage

[CVPR 2023 Highlight] InternImage: Exploring Large-Scale Vision Foundation Models with Deformable Convolutions

https://arxiv.org/abs/2211.05778

MIT License

2.47k stars 231 forks source link

Update: support DCNv4 in InternImage! #276

Closed zhiqi-li closed 7 months ago

zhiqi-li commented 7 months ago

We support DCNv4 in InternImage

with DCNv3: ./dist_test.sh configs/coco/mask_rcnn_internimage_t_fpn_1x_coco.py mask_rcnn_internimage_t_fpn_1x_coco.pth 8 --eval bbox
with DCNv4 ./dist_test.sh configs/coco/mask_rcnn_internimage_t_fpn_1x_coco_with_dcnv4.py mask_rcnn_internimage_t_fpn_1x_coco.pth 8 --eval bbox

zhiqi-li commented 7 months ago

Due to the special design of DCNv4, DCNv4 is slightly incompatible with InternImage and requires conversion before it can be used, which affects efficiency to a certain extent.