有人可以给我展示一个示例,该示例如何在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())