【发布时间】:2020-11-28 15:37:19
【问题描述】:
我正在尝试在 Google Colab 上训练自定义对象检测模型。我正在使用 Tensorflow gpu 2.2.0。
但是当我尝试在单元格 4 中导入库时(你可以在下面看到)我收到错误:AttributeError:模块'tensorflow.python.keras.utils.generic_utils'没有属性'populate_dict_with_module_objects'。
我尝试与 Tensorflow gpu 2.1.0 相同,但效果不佳。当我尝试在没有import tensorflow as tf 的情况下运行第 4 个单元格时,它运行时没有错误。这背后的原因是什么?无论如何我都需要解决这个问题。
(1) !pip install -U --pre tensorflow_gpu=="2.2.0"
(2) import os
import pathlib
#To Clone the tensorflow models repository if it doesn't already exist
if "models" in pathlib.Path.cwd().parts:
while "models" in pathlib.Path.cwd().parts:
os.chdir('..')
elif not pathlib.Path('models').exists():
!git clone --depth 1 https://github.com/tensorflow/models
(3) # To Install the Object Detection API
%%bash
cd models/research/
protoc object_detection/protos/*.proto --python_out=.
cp object_detection/packages/tf2/setup.py .
python -m pip install .
(4) import matplotlib
import matplotlib.pyplot as plt
import os
import random
import io
import imageio
import glob
import scipy.misc
import numpy as np
from six import BytesIO
from PIL import Image, ImageDraw, ImageFont
from IPython.display import display, Javascript
from IPython.display import Image as IPyImage
import tensorflow as tf
from object_detection.utils import label_map_util
from object_detection.utils import config_util
from object_detection.utils import visualization_utils as viz_utils
from object_detection.utils import colab_utils
from object_detection.builders import model_builder
%matplotlib inline
这是错误的图片:
【问题讨论】:
标签: python tensorflow google-colaboratory tensorflow2.0 object-detection-api