ZrrSkywalker / MonoDETR

[ICCV 2023] The first DETR model for monocular 3D object detection with depth-guided transformer
327 stars 31 forks source link

Turn the model to original version(running in 3090) #51

Open Ivan-Tang-3D opened 6 months ago

Ivan-Tang-3D commented 6 months ago

If u want to adapt the model to 3090, u could set the group_detr param in cfg to 1,and comment the lines of 467-473(about conditional) in the https://github.com/ZrrSkywalker/MonoDETR/blob/main/lib/models/monodetr/depthaware_transformer.py, then the model turns to the original version.

tingxuangao commented 6 months ago

did you mean the group_num parameter ? cuz i can't find 'group_detr'

Ivan-Tang-3D commented 5 months ago

Sorry, I made the mistake. Just the group_num parameter.

MarriedSenior commented 1 month ago

can you explain more in detail. i cant understand what you mean

my env is 3090. so this code is not working