Added normals to prediction DataObject

This commit is contained in:
Si11ium
2019-08-09 10:25:16 +02:00
parent 4e1fcdfd43
commit 8eb165f76c
7 changed files with 101464 additions and 2060 deletions

200
predict/clusters.txt Normal file
View File

@ -0,0 +1,200 @@
48
1.000000000000000000e+00
1 6
7.199833552042643747e-01 1.481056722005208437e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
5 6
3.199843406677246316e-01 1.547723388671875089e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
4.533166567484537834e-01 1.414390055338541563e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
3.866499900817871316e-01 1.614390055338541741e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
2.533176740010579242e-01 1.614390055338541741e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
3.866499900817871316e-01 1.481056722005208437e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
1.000000000000000000e+00
2 6
7.199833552042643747e-01 1.414390055338541563e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
6.533166885375976118e-01 1.481056722005208437e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
1.000000000000000000e+00
3.000000000000000000e+00
4 6
2.533176740010579242e-01 1.414390055338541563e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
3.199843406677246316e-01 1.347723388671874911e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
2.533176740010579242e-01 1.281056722005208259e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
3.866499900817871316e-01 1.281056722005208259e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
1.000000000000000000e+00
11 6
3.199843406677246316e-01 1.947723388671875000e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
4.533166567484537834e-01 1.947723388671875000e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
4.533166567484537834e-01 1.814390055338541696e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
5.199833234151204353e-01 2.014390055338541874e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
3.866499900817871316e-01 2.014390055338541874e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
5.199833234151204353e-01 2.147723388671875178e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
2.533176740010579242e-01 2.014390055338541874e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
3.199843406677246316e-01 1.814390055338541696e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
2.533176740010579242e-01 1.881056722005208348e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
4.533166567484537834e-01 2.081056722005208304e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
3.866499900817871316e-01 1.881056722005208348e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
8 6
3.199843406677246316e-01 2.214390055338541607e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
3.866499900817871316e-01 2.147723388671875178e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
4.533166567484537834e-01 2.081056722005208304e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
4.533166567484537834e-01 2.214390055338541607e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
5.199833234151204353e-01 2.014390055338541874e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
2.533176740010579242e-01 2.281056722005208481e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
3.866499900817871316e-01 2.014390055338541874e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
5.199833234151204353e-01 2.147723388671875178e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
1.000000000000000000e+00
4 6
7.199833552042643747e-01 1.947723388671875000e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
6.533166885375976118e-01 2.014390055338541874e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
7.199833552042643747e-01 2.081056722005208304e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
6.533166885375976118e-01 2.147723388671875178e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
1.000000000000000000e+00
1 6
1.866510073343912723e-01 2.281056722005208481e+00 5.645600001017252456e-01 -4.267321706433224227e-04 9.999984323090402860e-01 -1.718510726318397703e-03
1.000000000000000000e+00
1 6
5.199833234151204353e-01 2.281056722005208481e+00 5.645600001017252456e-01 -4.267321706433224227e-04 9.999984323090402860e-01 -1.718510726318397703e-03
1.000000000000000000e+00
1 6
6.533166885375976118e-01 1.747723388671875044e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
1.000000000000000000e+00
1 6
5.331766605377197266e-02 2.214390055338541607e+00 6.312266667683918975e-01 -9.993677590519846055e-01 -4.875568776990416931e-04 -3.555059008941406640e-02
1.000000000000000000e+00
1 6
1.199843327204386384e-01 2.147723388671875178e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
1.000000000000000000e+00
1 6
1.199843327204386384e-01 2.014390055338541874e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
1.000000000000000000e+00
2 6
5.331766605377197266e-02 2.081056722005208304e+00 6.312266667683918975e-01 -9.993677590519846055e-01 -4.875568776990416931e-04 -3.555059008941406640e-02
1.199843327204386384e-01 2.081056722005208304e+00 5.645600001017252456e-01 -9.993677590519846055e-01 -4.875568776990416931e-04 -3.555059008941406640e-02
1.000000000000000000e+00
1 6
1.199843327204386384e-01 1.614390055338541741e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
1.000000000000000000e+00
1 6
1.199843327204386384e-01 1.947723388671875000e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
5 6
3.199843406677246316e-01 2.281056722005208481e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
3.199843406677246316e-01 2.147723388671875178e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
3.866499900817871316e-01 2.214390055338541607e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
4.533166567484537834e-01 2.281056722005208481e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
2.533176740010579242e-01 2.214390055338541607e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
5 6
5.199833234151204353e-01 1.547723388671875089e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
3.866499900817871316e-01 1.681056722005208393e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
4.533166567484537834e-01 1.614390055338541741e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
4.533166567484537834e-01 1.481056722005208437e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
5.866500218709309600e-01 1.614390055338541741e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
3.000000000000000000e+00
1 6
5.331766605377197266e-02 1.881056722005208348e+00 6.312266667683918975e-01 -9.993677590519846055e-01 -4.875568776990416931e-04 -3.555059008941406640e-02
1.000000000000000000e+00
1 6
7.199833552042643747e-01 1.747723388671875044e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
1 6
5.866500218709309600e-01 1.747723388671875044e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
2 6
5.866500218709309600e-01 1.614390055338541741e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
5.199833234151204353e-01 1.547723388671875089e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
1.000000000000000000e+00
8 6
5.199833234151204353e-01 1.814390055338541696e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
1.866510073343912723e-01 1.881056722005208348e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
2.533176740010579242e-01 1.947723388671875000e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
3.199843406677246316e-01 1.881056722005208348e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
4.533166567484537834e-01 1.747723388671875044e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
5.866500218709309600e-01 1.881056722005208348e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
4.533166567484537834e-01 1.881056722005208348e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
3.866499900817871316e-01 1.814390055338541696e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
3.000000000000000000e+00
2 6
5.199833234151204353e-01 1.281056722005208259e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
5.866500218709309600e-01 1.347723388671874911e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
1.000000000000000000e+00
1 6
5.866500218709309600e-01 1.414390055338541563e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
3 6
6.533166885375976118e-01 2.147723388671875178e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
6.533166885375976118e-01 2.281056722005208481e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
7.199833552042643747e-01 2.214390055338541607e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
3.000000000000000000e+00
1 6
5.199833234151204353e-01 1.281056722005208259e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
3.000000000000000000e+00
6 6
3.866499900817871316e-01 1.347723388671874911e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
2.533176740010579242e-01 1.347723388671874911e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
3.199843406677246316e-01 1.414390055338541563e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
2.533176740010579242e-01 1.481056722005208437e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
2.533176740010579242e-01 1.614390055338541741e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
3.199843406677246316e-01 1.547723388671875089e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
1 6
1.199843327204386384e-01 1.814390055338541696e+00 5.645600001017252456e-01 -9.993677590519846055e-01 -4.875568776990416931e-04 -3.555059008941406640e-02
1.000000000000000000e+00
1 6
1.199843327204386384e-01 1.414390055338541563e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
3.000000000000000000e+00
1 6
1.199843327204386384e-01 1.281056722005208259e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
1 6
1.199843327204386384e-01 2.147723388671875178e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
3.000000000000000000e+00
1 6
1.199843327204386384e-01 1.347723388671874911e+00 5.645600001017252456e-01 -9.993677590519846055e-01 -4.875568776990416931e-04 -3.555059008941406640e-02
1.000000000000000000e+00
1 6
7.199833552042643747e-01 1.614390055338541741e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
1.000000000000000000e+00
1 6
7.199833552042643747e-01 2.281056722005208481e+00 6.312266667683918975e-01 -4.267321706433224227e-04 9.999984323090402860e-01 -1.718510726318397703e-03
1.000000000000000000e+00
1 6
5.866500218709309600e-01 1.881056722005208348e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
1 6
1.199843327204386384e-01 1.547723388671875089e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
1 6
1.866510073343912723e-01 1.747723388671875044e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
1 6
2.533176740010579242e-01 2.081056722005208304e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
3.000000000000000000e+00
1 6
7.199833552042643747e-01 1.281056722005208259e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
1 6
7.199833552042643747e-01 1.881056722005208348e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
1 6
1.199843327204386384e-01 1.747723388671875044e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
1.000000000000000000e+00
1 6
6.533166885375976118e-01 2.014390055338541874e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
1 6
1.199843327204386384e-01 1.481056722005208437e+00 5.645600001017252456e-01 -9.993677590519846055e-01 -4.875568776990416931e-04 -3.555059008941406640e-02
1.000000000000000000e+00
1 6
1.199843327204386384e-01 1.414390055338541563e+00 4.978933334350585938e-01 3.555141376242736823e-02 -1.702250673669388707e-03 -9.993663989359141686e-01
1.000000000000000000e+00
1 6
2.533176740010579242e-01 1.747723388671875044e+00 6.312266667683918975e-01 -3.555141376242736823e-02 1.702250673669388707e-03 9.993663989359141686e-01
1.000000000000000000e+00
1 6
5.866500218709309600e-01 2.281056722005208481e+00 6.312266667683918975e-01 -2.872351790370336957e-02 6.003545025065727403e-01 7.992180120838876523e-01
1.000000000000000000e+00
1 6
3.866499900817871316e-01 2.281056722005208481e+00 5.645600001017252456e-01 -4.267321706433224227e-04 9.999984323090402860e-01 -1.718510726318397703e-03

File diff suppressed because it is too large Load Diff

101251
predict/pointclouds/1_pc.xyz Normal file

File diff suppressed because it is too large Load Diff

View File

@ -227,7 +227,7 @@ def draw_clusters(clusters):
def draw_sample_data(sample_data, colored_normals = False):
cloud = o3d.PointCloud()
cloud.points = o3d.Vector3dVector(sample_data[:,:3])
cloud.points = o3d.Vector3dVector(sample_data[:, :3])
cloud.colors = \
o3d.Vector3dVector(label2color(sample_data[:, 6].astype(int)) if not colored_normals else sample_data[:, 3:6])
@ -243,7 +243,7 @@ sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/../') # add proj
parser = argparse.ArgumentParser()
parser.add_argument('--npoints', type=int, default=2048, help='resample points number')
parser.add_argument('--model', type=str, default='./checkpoint/seg_model_custom_30.pth', help='model path')
parser.add_argument('--model', type=str, default='./checkpoint/seg_model_custom_3.pth', help='model path')
parser.add_argument('--sample_idx', type=int, default=0, help='select a sample to segment and view result')
parser.add_argument('--headers', type=strtobool, default=True, help='if raw files come with headers')
parser.add_argument('--collate_per_segment', type=strtobool, default=True, help='whether to look at pointclouds or sub')
@ -260,7 +260,7 @@ if __name__ == '__main__':
print('Create data set ..')
dataset_folder = './data/raw/predict/'
pointcloud_file = './pointclouds/0_pc.xyz'
pointcloud_file = './pointclouds/1_pc.xyz'
# Load and pre-process point cloud
pcloud = pc.read_pointcloud(pointcloud_file)
@ -304,7 +304,7 @@ if __name__ == '__main__':
mode='predict',
root_dir='data',
npoints=opt.npoints,
refresh=False,
refresh=True,
collate_per_segment=opt.collate_per_segment,
has_variations=opt.has_variations,
headers=opt.headers