matanatz / pcnn

Point Convolutional Neural Networks by Extension Operators
56 stars 13 forks source link

Segmentation evaluation #3

Open Salingo opened 6 years ago

Salingo commented 6 years ago

Hello, It seems that the evaluation file for part segmentation is missing? Also, I find that the segmentation network and classification network are trained independently, I wonder if these two networks can be combined in the convolution stage, then the network can split for different tasks, just like the architecture of pointnet/pointnet++? Thanks.

Salingo commented 6 years ago

BTW, I found you mentioned that the network loss combined both classification and segmetation loss, can you provied this model.py file? Thanks very much!

# model.py defines both classification net and segmentation net, which share the common global feature extractor network.
# In model.get_loss, we define the total loss to be weighted sum of the classification and segmentation losses.
# Here, we only train for segmentation network. Thus, we set weight to be 1.0.