在 Colab 中找不到 ssd_mobile_net_v2 DNN 库

Question

我正在使用 Tensorflow 训练手部检测模型，但遇到了问题。

这是训练模型的代码：

!cd RealTimeObjectDetection/RealTimeObjectDetection && python Tensorflow/models/research/object_detection/model_main_tf2.py --model_dir=/content/RealTimeObjectDetection/RealTimeObjectDetection/Tensorflow/workspace/models/my_ssd_mobnet --pipeline_config_path=Tensorflow/workspace/models/my_ssd_mobnet/pipeline.config --num_train_steps=5000

我有这个错误：

Node: 'ssd_mobile_net_v2_fpn_keras_feature_extractor/model/Conv1/Conv2D'
DNN library is not found.
     [[{{node ssd_mobile_net_v2_fpn_keras_feature_extractor/model/Conv1/Conv2D}}]] [Op:__inference__dummy_computation_fn_15081]

Answer 1

据我所知，这与新版本的 TensorFlow (2.8.0) 有关。我能够通过以下方式解决问题 - https://github.com/tensorflow/models/issues/10479#issuecomment-1029626502