OpenPSG icon indicating copy to clipboard operation
OpenPSG copied to clipboard

Questions about the pretrained DETR models for PSGTR and PSGFormer

Open Yassin-fan opened this issue 1 year ago • 3 comments

Thanks for your work! I have successfully installed the framework and run the code (no errors reported). However, I have a similar problem to #33 :during training, the result is always zero. image

According to the logs you provided, I compared and found that my log section does not have anything related to model loading: image

I used the latest code and did not make any changes.

My questions are:: Have you ever encountered this problem?

Is the lack of weight loading content in my logs omitted in the later code or does it not appear due to model loading failure?

The most important question: In what way are the pre-trained models of DETR such as detr_pan_r50.pth and detr4psgformer_r50.pth obtained, and can we train them ourselves? What is the difference between them (in my understanding, both pre-trained models use their Encoder part for global information extraction)

Thanks!

Yassin-fan avatar May 17 '23 01:05 Yassin-fan