isl-org / isl-org/Open3D-ML

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

Open
#560 4 comments 0 reactions 0 assignees View on GitHub
question
Dominant language
Python
Stars
2.3k
Forks
365
Avg merge
4h 6m
Merged PRs (30d)
1

Description

### Checklist

- [X] I have searched for [similar issues](https://github.com/isl-org/Open3D-ML/issues).
- [X] I have tested with the [latest development wheel](http://www.open3d.org/docs/latest/getting_started.html#development-version-pip).
- [X] I have checked the [release documentation](http://www.open3d.org/docs/release/) and the [latest documentation](http://www.open3d.org/docs/latest/) (for `master` branch).

### 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:
```bash
python scripts/run_pipeline.py tf -c ml3d/configs/pointrcnn_kitti.yml --dataset.dataset_path --model.ckpt_path --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](https://user-images.githubusercontent.com/61607050/184861269-38ea11ec-60d6-480f-ab02-2e5a51e64b76.png)
![valid_cls_rpn](https://user-images.githubusercontent.com/61607050/184861315-ea47302e-dc46-4e5b-8a62-b603662947df.png)
![valid_reg_RPN](https://user-images.githubusercontent.com/61607050/184861323-5b4228c7-e5ed-4733-98fa-500738f214db.png)
![maps_RPN](https://user-images.githubusercontent.com/61607050/184861599-054547a3-01e8-4331-9806-a104e89f3d7d.png)

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.