代码之家 › 专栏 › 技术社区 › Austin

Fastai 1.0.x使用torchvision模型

torchvision fast-ai python

0

Austin · 技术社区 · 5 年前

create_cnn )? 我已经试过了,但是现在 RuntimeError: size mismatch

data = ImageDataBunch.from_name_re(path_img, 
                                   get_image_files(path/'images'), 
                                   r'/([^/]+)_\d+.jpg$', 
                                   ds_tfms=None, size=224, bs=16)

learn = create_cnn(data, models.alexnet, metrics=error_rate, 
                   custom_head=models.alexnet(num_classes=data.c).classifier)

print(learn.summary())

0 回复 | 直到 5 年前