pointnet2 icon indicating copy to clipboard operation
pointnet2 copied to clipboard

How to use the trained pointnet2_sem_seg model?

Open Coastline2018 opened this issue 5 years ago • 7 comments

@charlesq34 Thank you for your effort and contribution! After I trained the pointnet2_sem_seg model on the Sannet dataset, I want to load the trained model, and visualize the data, which is extracted from the scannet_test.pickle. However, I faced an error as follows: ValueError: No op named FarthestPointSample in defined operations. No matter the dataset is Scannet or the one made by myself, I face the same problem. Could you give me some tips to solve it? Thank you very much.

Coastline2018 avatar Aug 29 '18 16:08 Coastline2018

You have to import the custom librarys

# Import custom librarys BASE_DIR = os.path.dirname(os.path.abspath(__file__)) ROOT_DIR = os.path.dirname(BASE_DIR) TF_OPS_DIR = os.path.join(ROOT_DIR, 'tf_ops') sampling_module = tf.load_op_library(os.path.join(TF_OPS_DIR, 'sampling/tf_sampling_so.so')) grouping_module = tf.load_op_library(os.path.join(TF_OPS_DIR, 'grouping/tf_grouping_so.so')) interpolation_module = tf.load_op_library(os.path.join(TF_OPS_DIR, 3d_interpolation/tf_interpolate_so.so'))

julianstolp avatar Sep 04 '18 10:09 julianstolp

@Bierfass It works. Thank you so much!!!

Coastline2018 avatar Sep 04 '18 13:09 Coastline2018

@Coastline2018 I have trained the pointnet2_sem_seg model and get trained models. I want to load the model and test it with scannet_test.pickle. Also want to visualize the ground truth data and predict data. Is there any code prepared for this provided by the author? Do you write the test code by yourself? Thanks!!!

kxhit avatar Oct 08 '18 08:10 kxhit

@kxhit i also face this pozzle , have you ever solve this problem and have tested ?

kkyan1995 avatar Nov 05 '18 02:11 kkyan1995

Hey did u find the code?

akhilsantha7 avatar Apr 11 '19 17:04 akhilsantha7

@kxhit @kkyan1995 @Akhilreddy7777

did anyone ever figure out where the code is to test the train sem_seg model?

Naor92 avatar Oct 04 '19 21:10 Naor92

Hi @Coastline2018,

Could you please share the trained model pointnet2_sem_seg with us? We would appreciate it. My email is [email protected]

Thanks very much.

KevinYuk avatar Nov 22 '19 02:11 KevinYuk