Skip to content

Commit

Permalink
Merge pull request #3 from coocoo90/main
Browse files Browse the repository at this point in the history
fix S3 links; add topics;
  • Loading branch information
bryanyzhu authored Aug 26, 2022
2 parents c078fb0 + ff3e5da commit f610c97
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,23 +12,23 @@ This repo contains the supported code to reproduce spatio-temporal action detect

| Backbone | Pretrain | #view | mAP | FLOPs | config | model |
| :---: | :---: | :---: |:----:| :---: | :---: | :---: |
| CSN-50 | Kinetics-400 | 1 view | 27.2 | 78G | [config](configuration/TubeR_CSN50_AVA21.yaml) | [S3]([TBD](https://yzaws-data-log.s3.amazonaws.com/shared/TubeR_cvpr22/TubeR_CSN50_AVA21.pth)) |
| CSN-50 | Kinetics-400 | 1 view | 27.2 | 78G | [config](configuration/TubeR_CSN50_AVA21.yaml) | [S3](https://yzaws-data-log.s3.amazonaws.com/shared/TubeR_cvpr22/TubeR_CSN50_AVA21.pth) |
| CSN-50 (with long-term context) | Kinetics-400 | 1 view | 28.8 | 78G | [config](TBD) | Comming soon |
| CSN-152 | Kinetics-400+IG65M | 1 view | 29.7 | 120G | [config](configuration/TubeR_CSN152_AVA21.yaml) | [S3]([TBD](https://yzaws-data-log.s3.amazonaws.com/shared/TubeR_cvpr22/TubeR_CSN152_AVA21.pth)) |
| CSN-152 | Kinetics-400+IG65M | 1 view | 29.7 | 120G | [config](configuration/TubeR_CSN152_AVA21.yaml) | [S3](https://yzaws-data-log.s3.amazonaws.com/shared/TubeR_cvpr22/TubeR_CSN152_AVA21.pth) |
| CSN-152 (with long-term context) | Kinetics-400+IG65M | 1 view | 31.7 | 120G | [config](TBD) | Comming soon |


### AVA 2.2 Dataset

| Backbone | Pretrain | #view | mAP | FLOPs | config | model |
| :---: | :---: | :---: |:----:| :---: | :---: | :---: |
| CSN-152 | Kinetics-400+IG65M | 1 view | 31.1 | 120G | [config](configuration/TubeR_CSN152_AVA22.yaml) | [S3]([TBD](https://yzaws-data-log.s3.amazonaws.com/shared/TubeR_cvpr22/TubeR_CSN152_AVA22.pth)) |
| CSN-152 | Kinetics-400+IG65M | 1 view | 31.1 | 120G | [config](configuration/TubeR_CSN152_AVA22.yaml) | [S3](https://yzaws-data-log.s3.amazonaws.com/shared/TubeR_cvpr22/TubeR_CSN152_AVA22.pth) |
| CSN-152 (with long-term context) | Kinetics-400+IG65M | 1 view | 33.4 | 120G | [config](TBD) | Comming soon |

### JHMDB Dataset
| Backbone | #view | [email protected] | [email protected] | config | model |
| :---: | :---: | :---: | :---: | :---: | :---: |
| CSN-152 | 1 view | 87.4 | 82.3 | [config](configuration/Tuber_CSN152_JHMDB.yaml) | [S3]([TBD](https://yzaws-data-log.s3.amazonaws.com/shared/TubeR_cvpr22/TubeR_CSN152_JHMDB.pth)) |
| CSN-152 | 1 view | 87.4 | 82.3 | [config](configuration/Tuber_CSN152_JHMDB.yaml) | [S3](https://yzaws-data-log.s3.amazonaws.com/shared/TubeR_cvpr22/TubeR_CSN152_JHMDB.pth) |



Expand Down Expand Up @@ -85,7 +85,7 @@ python3 train_tuber_ava.py configuration/TubeR_CSN152_AVA21.yaml
## TODO
[ ]Add tutorial and pre-trained weights for TubeR with long-term memory

[ ] Add add weights for UCF24
[ ]Add weights for UCF24


## Citing TubeR
Expand Down

0 comments on commit f610c97

Please sign in to comment.