isl-org / Open3D-ML

An extension of Open3D to address 3D Machine Learning tasks
Other
1.76k stars 314 forks source link

Mismatch between labels and pointcloud when visualizing a custom dataset GT #461

Open Yinxq opened 2 years ago

Yinxq commented 2 years ago

I was trying to visualize the ground truth of a sementic segmentation dataset named Rellis-3D and encounted some problems:

I modified the following files according to theirs Semantic KITTI version:

ml3d/datasets/rellis.py
ml3d/vis/labellut.py
ml3d/datasets/_resources/rellis.yaml
ml3d/datasets/utils/rellis.yaml
ml3d/datasets/__init__.py
ml3d/configs/default_cfgs/rellis.yml

I think these files was customized properly, but in the visualizer, the labels and pointcloud were mismatched, for example, as the Figure shows: only "grass" and "fence" was visible, but I got pointcloud of "grass" and "person"; 2022-01-12 21-03-46屏幕截图

however, when i test the Semantic KITTI dataset, it works well.

Any help is greatly appreciated, thanks!

ssheorey commented 2 years ago

Hi @Yinxq the mapping between class label and class names for SemanticKITTI is here: https://github.com/isl-org/Open3D-ML/blob/5baf7223e9126e26feffc6f08d4b9ae728367dd1/ml3d/datasets/semantickitti.py#L112 You can modify this as appropriate for Rellis-3D.

Yinxq commented 2 years ago

Hi @Yinxq the mapping between class label and class names for SemanticKITTI is here:

https://github.com/isl-org/Open3D-ML/blob/5baf7223e9126e26feffc6f08d4b9ae728367dd1/ml3d/datasets/semantickitti.py#L112

You can modify this as appropriate for Rellis-3D.

Hi, @ssheorey the mapping between class label and class names for Rellis-3D is correct, the dataset file rellis.py is shown in the link: https://github.com/Yinxq/Open3D-ML/blob/master/ml3d/datasets/rellis.py#L108 I think my problem is similar to these issues: #225 #243 #409 , but #422 doesn't fix this problem in my case.

sanskar107 commented 2 years ago

@Yinxq this must be a problem from your side as all datasets supported by Open3D-ML works just fine. Can you upload the files you've changed so that we can test it ?

obeidahmad commented 6 months ago

@sanskar107 I have this same problem when visualizing non dataset data. Aka when I load data manually and use Visualizer().visualize() instead of Visualizer().visualize_dataset()

obeidahmad commented 6 months ago

Loading my data in a CustomDataset() Class fixed the problem. It's still weird however that it doesn't well support the visualization of custom data like in the documentation