Backbone自监督预训练

数据准备

无标签影像数据,越多越好。

模型训练

自监督模型选择SimSiam

import math
import torch
import torch.nn as nn
import torchvision
import numpy as np
import lightly
from lightly.models.modules.heads import 

猜你喜欢

转载自blog.csdn.net/weixin_42990464/article/details/125747581