Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

CodeCamp #43 Support PETR in 1.1 in projects #2175

Merged
merged 33 commits into from
Jan 5, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
33 commits
Select commit Hold shift + click to select a range
e820639
rebase
SekiroRong Dec 27, 2022
aca498f
petr3d-
SekiroRong Dec 27, 2022
f5596a5
petr3d-to-petr
SekiroRong Dec 27, 2022
65ff69d
delete_NormalizeMultiviewImage
SekiroRong Dec 27, 2022
06d684b
rename_PETR
SekiroRong Dec 27, 2022
b401698
rename_PETR
SekiroRong Dec 27, 2022
3e6bed0
fix_bug
SekiroRong Dec 27, 2022
7660bc0
fix_bug
SekiroRong Dec 27, 2022
06ab7dc
fix_bug
SekiroRong Dec 27, 2022
2ff66e9
fix_bug
SekiroRong Dec 27, 2022
0e72a91
fix_bug
SekiroRong Dec 27, 2022
2d5209f
fix_bug
SekiroRong Dec 27, 2022
3b5aaa8
fix_bug
SekiroRong Dec 27, 2022
43832c6
revise
SekiroRong Dec 27, 2022
e74655a
remove_builder
SekiroRong Dec 27, 2022
ce970bb
remove_builder
SekiroRong Dec 27, 2022
73bd987
remove_use_external
SekiroRong Dec 27, 2022
d70d1c0
remove_use_external
SekiroRong Dec 27, 2022
24b1c72
remove_PadMultiViewImage
SekiroRong Dec 27, 2022
7851682
remove_PadMultiViewImage
SekiroRong Dec 27, 2022
a46e7d3
remove-AddCamInfo
SekiroRong Dec 28, 2022
56d4821
remove-LidarBox3dVersionTransfrom
SekiroRong Dec 28, 2022
765dbb6
remove-LidarBox3dVersionTransfrom-and-AddCamInfo
SekiroRong Dec 28, 2022
9faa75b
fix__init__
SekiroRong Dec 28, 2022
b504b82
remove-redundent-config
SekiroRong Dec 29, 2022
2d72d17
code-polish
SekiroRong Dec 29, 2022
7f1df2e
remove-builder
SekiroRong Dec 29, 2022
15da09f
remove-builder
SekiroRong Dec 29, 2022
f1be572
remove-redundent-files
SekiroRong Dec 29, 2022
21acc17
replace-forward-train-and-test
SekiroRong Dec 29, 2022
f0af246
remove-redundent__init__
SekiroRong Dec 29, 2022
8c7cd00
remove_petr
SekiroRong Jan 4, 2023
7ee9757
remove-hierarchtecture
SekiroRong Jan 4, 2023
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
63 changes: 63 additions & 0 deletions projects/PETR/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,63 @@
# PETR

This is an README for `PETR`.

## Description

Author: @SekiroRong.
This is an implementation of *PETR*.

## Usage

<!-- For a typical model, this section should contain the commands for training and testing. You are also suggested to dump your environment specification to env.yml by `conda env export > env.yml`. -->

### Training commands

In MMDet3D's root directory, run the following command to train the model:

```bash
python tools/train.py projects/PETR/config/petr/petr_vovnet_gridmask_p4_800x320.py
```

### Testing commands

In MMDet3D's root directory, run the following command to test the model:

```bash
python tools/test.py projects/PETR/config/petr/petr_vovnet_gridmask_p4_800x320.py ${CHECKPOINT_PATH}
```

## Results

<!-- List the results as usually done in other model's README. [Example](https://github.com/open-mmlab/mmdetection3d/edit/dev-1.x/configs/fcos3d/README.md)
You should claim whether this is based on the pre-trained weights, which are converted from the official release; or it's a reproduced result obtained from retraining the model in this project. -->

This Result is trained by petr_vovnet_gridmask_p4_800x320.py and use [weights](https://drive.google.com/file/d/1ABI5BoQCkCkP4B0pO5KBJ3Ni0tei0gZi/view?usp=sharing) as pretrain weight.

| Backbone | Lr schd | Mem (GB) | Inf time (fps) | mAP | NDS | Download |
| :----------------------------------------------------------------------------------------------: | :-----: | :------: | :------------: | :--: | :--: | :----------------------: |
| [petr_vovnet_gridmask_p4_800x320](projects/PETR/configs/petr/petr_vovnet_gridmask_p4_800x320.py) | 1x | 7.62 | 18.7 | 38.3 | 43.5 | [model](<>) \| [log](<>) |

```
mAP: 0.3830
mATE: 0.7547
mASE: 0.2683
mAOE: 0.4948
mAVE: 0.8331
mAAE: 0.2056
NDS: 0.4358
Eval time: 118.7s
Per-class results:
Object Class AP ATE ASE AOE AVE AAE
car 0.567 0.538 0.151 0.086 0.873 0.212
truck 0.341 0.785 0.213 0.113 0.821 0.234
bus 0.426 0.766 0.201 0.128 1.813 0.343
trailer 0.216 1.116 0.227 0.649 0.640 0.122
construction_vehicle 0.093 1.118 0.483 1.292 0.217 0.330
pedestrian 0.453 0.685 0.293 0.644 0.535 0.238
motorcycle 0.374 0.700 0.253 0.624 1.291 0.154
bicycle 0.345 0.622 0.262 0.775 0.475 0.011
traffic_cone 0.539 0.557 0.319 nan nan nan
barrier 0.476 0.661 0.279 0.142 nan nan
```
Loading