Skip to content

agentoo7/SOLOv2

 
 

Repository files navigation

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

Results(Model is still in training, will update result once finished.)

4 epoch AP: 0.281

Visualization(1 epoch)

About

SOLOv2: Dynamic, Faster and Stronger

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 88.0%
  • Cuda 7.8%
  • C 4.1%
  • Shell 0.1%