mmselfsup/docs/zh_cn/model_zoo.md

12 KiB

Model Zoo

All models and benchmarks results are recorded below.

Pre-trained models

Algorithm Config Download Remark
BYOL config model 8xb32 / Accumulate 16
DeepCLuster config model
DenseCL config model
MoCo v2 config model
NPID config model
config model Ensure Negative
Online DeepCluster config model
Relative Location config model
Rotation Prediction config model
SimCLR config model
SimSiam config model Epoch 100
config model Epoch 200
SwAV config model

Benchmarks

In following tables, we only displayed ImageNet Linear Evaluation, COCO17 Object Detection and PASCAL VOC12 Aug Segmentation, you can click the model name above to get the comprehensive benchmark results.

ImageNet Linear Evaluation

If not specified, we use linear evaluation setting from MoCo. Or the settings is mentioned in Remark.

Algorithm Config Download Remark Top-1 (%)
BYOL config model 8xb32 / Accumulate 16 67.68
DeepCLuster config model
DenseCL config model
MoCo v2 config model 67.56
NPID config model
config model Ensure Negative
Online DeepCluster config model
Relative Location config model 39.65
Rotation Prediction config model 44.35
SimCLR config model
SimSiam config model Epoch 100 / SimSiam paper setting
config model Epoch 200 / SimSiam paper setting
SwAV config model SwAV paper setting

COCO17 Object Detection

In COCO17 Object detection task, we choose the evluation protocol from MoCo, with Mask-RCNN architecture, the results below are trained with the same config.

Algorithm Config Download Remark AP (Box) AP (Mask)
BYOL config model 8xb32 / Accumulate 16
DeepCLuster config model
DenseCL config model
MoCo v2 config model
NPID config model
config model Ensure Negative
Online DeepCluster config model
Relative Location config model
Rotation Prediction config model
SimCLR config model
SimSiam config model Epoch 100 36.9 33.2
config model Epoch 200
SwAV config model

Pascal VOC12 Aug Segmentation

In Pascal VOC12 Aug Segmentation task, we choose the evluation protocol from MoCo, with FCN architecture, the results below are trained with the same config.

Algorithm Config Download Remark mIOU
BYOL config model 8xb32 / Accumulate 16
DeepCLuster config model
DenseCL config model
MoCo v2 config model
NPID config model
config model Ensure Negative
Online DeepCluster config model
Relative Location config model
Rotation Prediction config model
SimCLR config model
SimSiam config model Epoch 100
config model Epoch 200
SwAV config model