search

prs-eth / ForAINet

official source code for paper entitled "Automated forest inventory: analysis of high-density airborne LiDAR point clouds with 3D deep learning"
50 stars 9 forks source link

ForAINet支持自定义数据集吗（单类别） #3

Open TstMua opened 6 months ago

TstMua commented 6 months ago

ForAINet支持自定义数据集吗（单类别）

bxiang233 commented 6 months ago

ForAINet支持自定义数据集吗（单类别）

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

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

TstMua commented 6 months ago

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

bxiang233 commented 6 months ago

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

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

TstMua commented 6 months ago

好的，谢谢