Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
jz462 committed Aug 16, 2019
1 parent b0619a6 commit a1fc3f2
Showing 1 changed file with 9 additions and 10 deletions.
19 changes: 9 additions & 10 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,16 +12,15 @@ This is a PyTorch implementation for [Graphical Contrastive Losses for Scene Gra
We have created a branch for a version supporting pytorch1.0! Just go to the [pytorch1_0](https://github.com/NVIDIA/ContrastiveLosses4VRD/tree/pytorch1_0) branch and check it out!

## Benchmarking
| Method | SGDET@20 | SGDET@50 | SGDET@100 |
| :--- | :----: | :----: | :----: |
| Associative Embedding \[1\] | 6.5 | 8.1 | 8.2 |
| Message Passing \[2\] | 14.6 | 20.7 | 24.5 |
| Frequency \[3\] | 17.7 | 23.5 | 27.6 |
| Frequency+Overlap \[4\] | 20.1 | 26.2 | 30.1 |
| MotifNet \[5\] | 21.4 | 27.2 | 30.3 |
| RelDN, VGG16, w/o contrastive losses | 20.8 | 28.1 | 32.5 |
| RelDN, VGG16, full | 21.1 | 28.3 | 32.7 |
| RelDN, ResNext-101-FPN, full | 22.5 | 31.0 | 36.7 |
| Method | Backbone | SGDET@20 | SGDET@50 | SGDET@100 |
| :--- | :----: | :----: | :----: | :----: |
| Frequency \[1\] | VGG16 | 17.7 | 23.5 | 27.6 |
| Frequency+Overlap \[1\] | VGG16 | 20.1 | 26.2 | 30.1 |
| MotifNet \[1\] | VGG16 | 21.4 | 27.2 | 30.3 |
| Graph-RCNN \[2\] | Res-101 | 19.4 | 25.0 | 28.5 |
| RelDN, w/o contrastive losses | VGG16 | 20.8 | 28.1 | 32.5 |
| RelDN, full | VGG16 | 21.1 | 28.3 | 32.7 |
| RelDN, full | ResNext-101-FPN | 22.5 | 31.0 | 36.7 |


## Cloning
Expand Down

0 comments on commit a1fc3f2

Please sign in to comment.