search

skyhehe123 / SA-SSD

SA-SSD: Structure Aware Single-stage 3D Object Detection from Point Cloud (CVPR 2020)
492 stars 106 forks source link

Help needed in visualizing results #80

Open kaushik333 opened 3 years ago

kaushik333 commented 3 years ago

@skyhehe123 I am trying to visualize the results after performing training. However I do not see any code-instructions to do so. Can you please point me towards how you visualized your results? I came across kittiviewer in the pointPillars code-base. But this code is not generic and is tied to the repo. So can you please help me regarding this? Thanks in advance.

skyhehe123 commented 3 years ago

@kaushik333 The drawing function for both image and point cloud are implemented in https://github.com/skyhehe123/SA-SSD/blob/2d75c973af65453186bd9242d7fa5e62dc44ec03/mmdet/datasets/kitti_utils.py. To visualize point cloud, please install mayavi package.

kaushik333 commented 3 years ago

Thank you for you response. I am having trouble running mayavi. I am trying to visualize the results in a jupyter notebook on a remote server. But mayavi just isnt working for me. The library loads, but the jupyter cell doesnt respond at all i.e. keeps running indefinitely. mlab.figure() is what gives me this issue.

Is there a way to save and view mayavi figures later?

hitxing commented 3 years ago

Thank you for you response. I am having trouble running mayavi. I am trying to visualize the results in a jupyter notebook on a remote server. But mayavi just isnt working for me. The library loads, but the jupyter cell doesnt respond at all i.e. keeps running indefinitely. mlab.figure() is what gives me this issue.

Is there a way to save and view mayavi figures later?

Hi, I am also trying to visualize the results, have you already finished it? If so, could you please point me towards how to visualized your results?

kaushik333 commented 3 years ago

This is the repository I used to visualize my code : https://github.com/kuixu/kitti_object_vis

More specifically this : https://github.com/kuixu/kitti_object_vis/tree/master/jupyter

In my environment I was not able to get the dependencies to work. mayavi has some weird version dependencies which I was ultimately not able to figure out. So, I used this docker image and tried it inside that.

https://hub.docker.com/layers/pytorch/pytorch/1.5-cuda10.1-cudnn7-devel/images/sha256-449d5b98aa911955f55cb3ab413b81b34861ab5f4244519099328df282275690?context=explore

as mentioned in the tutorial page, make sure to set export ETS_TOOLKIT='null' everytime. There are a couple of places where they suggest setting it to qt4, pyqt5 etc. But none of that seemed to work for me.

If you are also running it on a remote system like me and if you need more help, please let me know.

hitxing commented 3 years ago

@kaushik333 Thanks a lot! I'm sorry not reply you in time since I'm a bit busy these days，If possible，maybe we communicate with WeChat is more convenient, my WeChat is HLX18774475803.