python - Fastai 1.0.x 使用 torchvision 模型
问题描述
有人可以向我展示一个示例,说明如何在 fastai 中使用 torchvision 模型而不进行修改(即没有进行网络修改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())
解决方案
推荐阅读
- google-api - 我可以/应该隐藏错误 403: org_internal in google login 的详细信息吗?
- date - Windows批处理文件检查时间和日期然后重命名
- node.js - 确定最佳变量以获得最高回报的最快方法
- python-3.x - Python 3 中 bisect_left 中的索引超出范围
- javascript - Javascript post请求函数返回未定义
- mysql - PHP CodeIgniter 和 JQuery AJAX 使用 jquery 追加表行并将所有表数据插入数据库
- spring-boot - SpringBoot RedirectAttributes 未显示在百里香叶中
- r - R中的闰年一些输出是错误的
- asp.net-core - .net 核心 webapp 和控制台应用程序是否有一致的方法来确定数据文件文件夹的路径?
- php - php-将记录插入数据库