Mediapipe Model Maker导入object_detector失败问题求助

环境信息 #

当前使用的依赖版本：

Python 3.10.13
Tensorflow 2.10.1
Mediapipe 0.9.0.1
Mediapipe-model-maker 0.1.0.2

报错情况 #

执行导入代码时：

import os
import json
import tensorflow as tf
import matplotlib.pyplot as plt
from matplotlib import patches, text, patheffects
from collections import defaultdict
import math

# 问题行
from mediapipe_model_maker import object_detector

assert tf.__version__.startswith('2')

触发如下错误：

from mediapipe_model_maker import object_detector
ImportError: cannot import name 'object_detector' from 'mediapipe_model_maker' (C:\Users\username\Anaconda3\envs\object_detection_mediapipe\lib\site-packages\mediapipe_model_maker\__init__.py)

排查细节 #

查看报错路径下的__init__.py文件，内容如下：

# content of the __init__.py file \Anaconda3\envs\object_detection_mediapipe\lib\site-packages\mediapipe_model_maker\__init__.py

from mediapipe_model_maker.python.core.utils import quantization
from mediapipe_model_maker.python.vision import image_classifier
from mediapipe_model_maker.python.vision import gesture_recognizer
from mediapipe_model_maker.python.text import text_classifier

可见该版本的mediapipe-model-maker并未在顶级模块导出object_detector。

解决方案 #

方案1：修改导入路径 #

旧版本的mediapipe-model-maker中，object_detector属于vision子模块的深层路径，将导入语句替换为：

from mediapipe_model_maker.python.vision import object_detector

方案2：升级mediapipe-model-maker版本 #

0.1.0.2版本未在顶级模块封装object_detector的导入，建议升级到兼容的新版本，执行以下命令：

pip install --upgrade mediapipe-model-maker==0.2.0.1

升级后即可使用原导入语句from mediapipe_model_maker import object_detector，注意该版本与TensorFlow 2.10.1兼容。

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