Need help with SOLOv2?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

Epiphqny
172 Stars 23 Forks Apache License 2.0 32 Commits 6 Opened issues

Description

SOLOv2: Dynamic, Faster and Stronger, achives 39.5mAP on coco test-dev (36 epochs result)

Services available

!
?

Need anything else?

Contributors list

# 255,384
Python
C++
Shell
instanc...
30 commits

SOLOv2

The code is an unofficial pytorch implementation of SOLOv2: Dynamic, Faster and Stronger

Install

Please check SOLOv1 for installation instructions.

Training

Follows the same way as SOLOv1.

single GPU:

python tools/train.py configs/solov2/solov2_r101_3x.py
multi GPU (for example 8):
./tools/dist_train.sh configs/solov2/solov2_r101_3x.py 8

Weights

Trained model can be download in Google drive BaiduYun 提取码: qw4e

Results

After training 36 epochs(3x) on the coco dataset using the resnet-101 backbone, the mAP is 39.5 on COCO test-dev2017 dataset. In the original paper, the model achieves 39.7 after 72 epochs(6x).

Visualization(1 epoch)

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.