Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ForAINet支持自定义数据集吗(单类别) #3

Open
TstMua opened this issue May 21, 2024 · 4 comments
Open

ForAINet支持自定义数据集吗(单类别) #3

TstMua opened this issue May 21, 2024 · 4 comments

Comments

@TstMua
Copy link

TstMua commented May 21, 2024

ForAINet支持自定义数据集吗(单类别)

@bxiang233
Copy link
Collaborator

ForAINet支持自定义数据集吗(单类别)

哈喽,支持自定义数据集的,只是你需要修改相应的代码,比如:
PointCloudSegmentation/torch_points3d/datasets/panoptic
和PointCloudSegmentation/torch_points3d/datasets/segmentation
下面你需要自定义你的数据集的文件,config file也要做对应修改。

不是很懂你说的单类别是指什么?如果你只想分树和非树的点的话,可以参考我们之前的工作:
https://github.com/prs-eth/PanopticSegForLargeScalePointCloud

@TstMua
Copy link
Author

TstMua commented May 22, 2024

嗨,我想将树上的苹果分割出来。数据集格式应该遵循什么风格的呀?我只有txt的点云数据。但我在数据集的制作上遇到了困难。

@bxiang233
Copy link
Collaborator

嗨,我想将树上的苹果分割出来。数据集格式应该遵循什么风格的呀?我只有txt的点云数据。但我在数据集的制作上遇到了困难。

你可以参考一下for-instance的数据,我们用的.ply文件格式,每行是x y z labels。但是.txt的也可以啊,你只用改一下dataloader的部分就可以了。

@TstMua
Copy link
Author

TstMua commented May 22, 2024

好的,谢谢

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants