This repo contains the supported code and configuration files of our final project. It is forked from https://github.com/SwinTransformer/Swin-Transformer-Object-Detection.
We used Swin-T model (the first row) in the following table.
Backbone | Pretrain | Lr Schd | box mAP | mask mAP | #params | FLOPs | config | log | model |
---|---|---|---|---|---|---|---|---|---|
Swin-T | ImageNet-1K | 1x | 43.7 | 39.8 | 48M | 267G | config | github/baidu | github/baidu |
Swin-T | ImageNet-1K | 3x | 46.0 | 41.6 | 48M | 267G | config | github/baidu | github/baidu |
Swin-S | ImageNet-1K | 3x | 48.5 | 43.3 | 69M | 359G | config | github/baidu | github/baidu |
For our config files, you can refer to config/food_recog and config/sw_test
For all the codes and data used in search engine module, you can refer to folder search_engine