search

SJingjia / PlaneRecTR

PlaneRecTR: Unified Query Learning for 3D Plane Recovery from a Single View
Apache License 2.0
24 stars 1 forks source link

inference on single image? #2

Closed aroop closed 5 months ago

aroop commented 6 months ago

Hi, Thank you for the great work. Would you please share the inference script of PlaneRecTR? I would like to test your work on my own image. Thanks!

SJingjia commented 6 months ago

Hi, thanks for your interest in our work. I'll upload the inference demo as soon, and If you need it urgently, you can also refer to https://github.com/facebookresearch/Mask2Former/blob/main/demo/demo.py, https://github.com/facebookresearch/detectron2/blob/master/demo/demo.py

IresAtLetink commented 5 months ago

Hi @SJingjia - thank you for you work!

I've tried to setup a script to setup inference based on the demo files with mixed result. Using the default config and the r50 pretrained data I'm getting a tensor size error on the inferred_seg_depth calculation. I'm having difficulty debugging - any pointers for me where to look?

SJingjia commented 5 months ago

Hi @IresAtLetink @aroop I've uploaded an inference demo you can try it out

aroop commented 5 months ago

Thanks @SJingjia

Verified that demo is working