fastai使用efficinentnet

最新推荐文章于 2020-11-12 18:52:43 发布

追梦小狂魔

最新推荐文章于 2020-11-12 18:52:43 发布

阅读量341

点赞数

分类专栏： fastai

本文链接：https://blog.csdn.net/qq_15557299/article/details/105337922

版权

fastai 专栏收录该内容

8 篇文章 1 订阅

订阅专栏

!pip install efficientnet-pytorch

from fastai import *
from fastai.vision import *
from efficientnet_pytorch import EfficientNet

path = untar_data(URLs.PETS)
path_anno = path/'annotations'
path_img = path/'images'
fnames = get_image_files(path_img)
pat = r'/([^/]+)_\d+.jpg$'

data = ImageDataBunch.from_name_re(path_img, fnames, pat, ds_tfms=get_transforms(), size=224, bs=32
                                  ).normalize(imagenet_stats)

model = EfficientNet.from_name('efficientnet-b0')
model._fc = nn.Linear(1280, data.c)
learn = Learner(data, model)
#b7 2560