SelfSup:基于cvpods的自我监督方法的集合 源码
自我支持 自我监督方法的集合(MoCo系列,SimCLR, SiMo ,BYOL,SimSiam,SwAV,PointContrast等)。 开始使用 按照说明安装cvpods。 从安装cvpods。 准备数据集 cd cvpods ln -s /path/to/your/ImageNet datasets/imagenet 训练自己的模型 cd /path/to/your/SelfSup/examples/simclr/simclr.res50.scratch.imagenet.224size.256bs.200e # pre-train pods_train --num-gpus 8 # convert to weights python convert.py simclr.res50.scratch.imagenet.224size.256bs.200e/log/model_fin
下载地址
用户评论