本文详细介绍了在Python测试代码中，如何使用importlib.import_module和getattr来正确加载网络定义。强调了当训练脚本名称更改时，相应modulePath和moduleName也需要同步更新，否则会导致预测结果出错，无法正确处理图像。对于依赖网络定义的预测流程，理解这一加载机制至关重要。
转载地址：https://bbs.huaweicloud.com/forum/thread-81062-1-1.html
作者：呼哈呼哈
需要注意测试代码中是通过import以及getattr来加载网络的，如果修改了训练脚本的名称，对应的modulePath名称也需要修改
# load the network definition
module = importlib.import_module(modulePath)
network = getattr(module, moduleName)()
否则无法加载网络，预测结果仍为32x32像素的图片，无法进行预测！
扫一扫
1416
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
