search

jerpelhan / DAVE

MIT License
42 stars 4 forks source link

Detection Stage Encoded Features #14

Open jwahnn opened 3 months ago

jwahnn commented 3 months ago

arch

I am having trouble finding which part of the code represents the features of the objects in the images that are fed into the decoder. I don't think such detail is in the demo, but I cannot find it in other files as well. Do you mind pointing me to this place?

And, it would be greatly appreciated if you could point me towards the verification stage code as well.

Thanks in advance!

jerpelhan commented 3 months ago

Sorry, I don't quite understand your question. Could you clarify what you mean by "features of the objects in the image that are fed into the decoder"?

The entire pipeline is in demo.py. When model(img, bboxes=bboxes) is called, it triggers the forward function in dave.py. The verification stage is clearly marked with a comment in the same file to help you navigate.