Fastai 1.0.x使用Torchvision模型

时间:2019-02-21 20:40:41

标签: python fast-ai torchvision

有人可以给我展示一个示例,该示例如何在Fastai中使用未经修改(即未经create_cnn进行网络修改的火炬视觉模型)吗?我已经尝试过了,但是得到RuntimeError: size mismatch。我认为alexnet输入大小应该为224,所以不确定如何调试。

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 个答案:

没有答案