OpenVINO中结合person-detection-0201与自定义分类模型遇输出兼容问题求助

背景信息 #

• 使用OpenVINO 2021.4版本，将Keras模型转为saved_model.pb格式后，再转换为IR格式（xml和bin文件）
• 用开发工具中的hello_classification.py测试自定义分类模型运行正常：单张图片输入时，class id 0输出概率100%，class id 1输出概率0%（自定义标签定义：0代表“戴帽子的人”，1代表“未戴帽子的人”）

问题详情 #

尝试将上述自定义分类模型与OpenVINO公开的person-detection-0201模型结合使用时，运行代码后出现“unsupported model output”错误。目前不清楚如何将这两个模型结合，实现判断人物是否戴帽子的功能，是否需要自定义或修改原object detection.py？

期望实现流程 #

1. 先用person-detection-0201模型检测出图像中的人物目标
2. 对检测出的人物区域进行是否戴帽子的二分类判断

内容的提问来源于stack exchange，提问作者newbieLife