Skip to content

Commit

Permalink
Merge pull request GAP-LAB-CUHK-SZ#1 from caoyue10
Browse files Browse the repository at this point in the history
Update README.md
  • Loading branch information
zeliu98 authored Apr 2, 2021
2 parents 0876247 + 33207d5 commit b445e2d
Showing 1 changed file with 7 additions and 9 deletions.
16 changes: 7 additions & 9 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,7 +1,8 @@
# Group-Free 3D Object Detection via Transformers

By [Ze Liu](https://github.com/zeliu98), [Zheng Zhang](https://github.com/stupidZZ)
, [Yue Cao](https://github.com/caoyue10), [Han Hu](https://github.com/ancientmooner), [Xin Tong](http://www.xtong.info/)
By [Ze Liu](https://github.com/zeliu98), [Zheng Zhang](https://github.com/stupidZZ), [Yue Cao](https://github.com/caoyue10), [Han Hu](https://github.com/ancientmooner), [Xin Tong](http://www.xtong.info/).

This repo is the official implementation of ["Group-Free 3D Object Detection via Transformers"](https://arxiv.org/abs/2104.).

![teaser](doc/teaser.png)

Expand All @@ -11,9 +12,6 @@ By [Ze Liu](https://github.com/zeliu98), [Zheng Zhang](https://github.com/stupid

## Introduction

This repo is the official implementation
of ["Group-Free 3D Object Detection via Transformers"](https://arxiv.org/abs/2104.).

Recently, directly detecting 3D objects from 3D point clouds has received increasing attention. To extract object
representation from an irregular point cloud, existing methods usually take a point grouping step to assign the points
to an object candidate so that a PointNet-like network could be used to derive object features from the grouped points.
Expand Down Expand Up @@ -42,7 +40,7 @@ scripts on ScanNet and SUN RGB-D.

## Main Results

### Scannet V2
### ScanNet V2

|Method | backbone | [email protected] | [email protected] | Model |
|:---:|:---:|:---:|:---:|:---:|
Expand All @@ -58,7 +56,7 @@ scripts on ScanNet and SUN RGB-D.
| Ours(L12, O256) | PointNet++w2× |68.8 (68.3*) | 52.1 (51.1*) |[model](https://drive.google.com/file/d/1V6sFLFcqsp7YJ3-9AV2NqUhEGVkuNGWT/view?usp=sharing)|
| Ours(L12, O512) | PointNet++w2× | 69.1 (68.8*) |52.8 (52.3*) |[model](https://drive.google.com/file/d/16NAEZqxPdBkxW7GGKGHe4-nDtfqL1htE/view?usp=sharing)|

### SUNRGBD
### SUN RGB-D

|Method | backbone | inputs | [email protected] | [email protected] | Model |
|:---:|:---:|:---:|:---:|:---:|:---:|
Expand Down Expand Up @@ -96,7 +94,7 @@ For ScanNet, follow the [README](./scannet/README.md) under the `scannet` folder

## Usage

### Scannet
### ScanNet

For `L6, O256` training:

Expand Down Expand Up @@ -170,7 +168,7 @@ python eval_avg.py --num_point 50000 --width 2 --num_decoder_layers 12 --num_tar
--dataset scannet --data_root <data directory> [--dump_dir <dump directory>]
```

#### SUNRGBD
#### SUN RGB-D

For `L6, O256` training:

Expand Down

0 comments on commit b445e2d

Please sign in to comment.