Open3D-ML icon indicating copy to clipboard operation
Open3D-ML copied to clipboard

How is the pretrained PointRCNN model supposed to be used? + Training Issues PointRCNN

Open remunds opened this issue 3 years ago • 4 comments

Checklist

My Question

Hi,

I am using the following versions:

  • tensorflow: 2.5.3
  • open3d: 0.15.2 (from pip), as well as a local git clone of the current master branch

I am currently trying to get PointRCNN Object Detection (TF) to work. For that I downloaded the pretrained weights and trying to validate the model using the following command:

python scripts/run_pipeline.py tf -c ml3d/configs/pointrcnn_kitti.yml --dataset.dataset_path <dataset_path> --model.ckpt_path <pretrained_path/ckpt-1> --split test --mode RCNN/RPN

I tried both modes. When in RPN mode, cls loss is at around 1.1 and reg loss at around 7.7. However, validation_MAPs are always 0.0 in all cases. In RCNN mode, the validation losses are around 0.0 and the validation_map's are again 0.0 in all cases.

So what mode has the pretrained PointRCNN been trained on? How should it be used?

Also, when I try to train PointRCNN by myself (RPN), the same issue with always zero validation_MAPs exists. Even after 50+ epochs, they stay at 0.0 in all cases. See the plots of the training and validation of the RPN below. Remember, this is with default settings.

train_losses_RPN valid_cls_rpn valid_reg_RPN maps_RPN

remunds avatar Aug 16 '22 10:08 remunds

So i checked the pretrained weights of PointRCNN (for tensorflow) using self.ckpt.restore(ckpt_path).assert_existing_objects_matched() and i get lots of warnings that the weights were not loaded properly. Was the default config or the model of PointRCNN changed after uploading the weights? That would probably be the easiest explanation (since using saved checkpoints from my own training worked and the pointpillar weights work too).

This means that the currently linked weights are not up to date. Would you be able to update these?

remunds avatar Sep 14 '22 14:09 remunds

Hi, did you solve this?

Jeremy26 avatar Feb 02 '23 17:02 Jeremy26

Nope, i switched to using PointPillars.

remunds avatar Feb 16 '23 17:02 remunds

@Jeremy26, did you find a solution? I'm exactly the same issue. Training results always in 0.

jvzacchi avatar Feb 27 '23 15:02 jvzacchi