close

下表為ResNet系列結構:

 

下面為ResNet系列第一種寫法(官方預設結構):

import torchvision.models as models

resnet18 = models.resnet18()

resnet34 = models.resnet34()

resnet50 = models.resnet50()

resnet101 = models.resnet101()

resnet152 = models.resnet152()

可以導入預訓練的model權重, 只要在()內填入pretrained參數即可, 如下:

resnet18 = model.resnet18(pretrained=True)

 

下面為Resnet18輸出的結果:

arrow
arrow
    文章標籤
    resnet resnet18 resnet34 resnet50 resnet101 resnet152 python pytorch
    全站熱搜

    楓綺 發表在 痞客邦 留言(0) 人氣()

    E-mail轉寄