Skip to content

Commit

Permalink
update readme (add speed with torchscript)
Browse files Browse the repository at this point in the history
  • Loading branch information
wangyingming committed Sep 30, 2021
1 parent 7b1f11c commit f688fb0
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -22,14 +22,13 @@ We also propose an attention variant RCDA to reduce the memory cost for high-res
| SMCA | multi-level | 50 | 43.7 | 152 | 10 |
| Deformable DETR | multi-level | 50 | 43.8 | 173 | 15 |
| Conditional DETR | DC5 | 50 | 43.8 | 195 | 10 |
| Anchor DETR | DC5 | 50 | 44.2 | 151 | 16 |
| Anchor DETR | DC5 | 50 | 44.2 | 151 | 16 (19) |


*Note:*
1. The results are based on ResNet-50 backbone.
2. Inference speeds are measured on NVIDIA Tesla V100 GPU.
3. DETR gets 12 FPS with torchscript optimization and 10 FPS without it.
The others do not enable the torchscript.
3. The values in parentheses of the Infer Speed indicate the speed with torchscript optimization.


## Model
Expand Down

0 comments on commit f688fb0

Please sign in to comment.