本文針對PointNet++強大的三維點云分類功能,詳細講解怎么訓練自己的數(shù)據(jù)集,在此之前,需要確保已經(jīng)能夠跑通源碼的訓練和測試,如果沒有,請參考PointNet++的源碼運行。
數(shù)據(jù)放置

1.1. 在mytensor_shape_names.txt中配置自己的分類,以及分類名稱:

1.2. 在filelist.txt中填入對應的不同類別的文件夾名/數(shù)據(jù)文件名,這里的文件夾名最好跟自己的分類類別一致

1.3. 然后在mytensor_train.txt和mytensor_test.txt中依次輸入訓練和測試的數(shù)據(jù)文件名稱
我的訓練集就是測試集,所以是一樣的

1.4 實際的再次采集的測試數(shù)據(jù)

代碼修改
2.1 訓練時的代碼修改
主要修改batch_size,和類別數(shù)量

在數(shù)據(jù)加載模塊中,將自己的配置文件放進去

然后就可以開始訓練
2.2 測試時代碼修改
測試時修改較多,主要是在數(shù)據(jù)加載類中,可直接查看代碼
運行結果查看
查看自己的分類是否正確,可直接查看打印出來的pred_choice

4. 源碼下載
可前往百度網(wǎng)盤下載,測試,有任何問題可以私信我
鏈接:https://pan.baidu.com/s/1iuqHQg_w6kq6JzmZjaIbaw 文章來源:http://www.zghlxwxcb.cn/news/detail-808202.html
提取碼:ng3h文章來源地址http://www.zghlxwxcb.cn/news/detail-808202.html
到了這里,關于PointNet++訓練自己的數(shù)據(jù)集(附源碼)的文章就介紹完了。如果您還想了解更多內容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關文章,希望大家以后多多支持TOY模板網(wǎng)!