Tensorflow2对象检测API：Numpy版本错误

我遵循了“训练自定义对象检测器”教程（https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/training.html）

运行脚本以继续训练预训练模型时：

python model_main_tf2.py --model_dir=models/my_ssd_resnet50_v1_fpn --pipeline_config_path=models/my_ssd_resnet50_v1_fpn/pipeline.config

（在这里找到：https : //github.com/tensorflow/models/blob/master/research/object_detection/model_main_tf2.py）

使用不同的 numpy 版本，我收到以下错误。

场景#1：

• 张量流：2.2.0
• 麻木：1.20.0-1

NotImplementedError: Cannot convert a symbolic Tensor (cond_2/strided_slice:0) to a numpy array. This error may indicate that you're trying to pass a Tensor to a NumPy call, which is not supported

我在网上查看，它建议将 numpy 版本降级 < 1.20.0（NotImplementedError: Cannot convert a Symbolic Tensor (2nd_target:0) to a numpy array）。请注意，对于 tensorflow 2.2.0，numpy 版本必须 >= 1.19.2。

场景#2：

• 张量流：2.2.0
• 麻木：1.19.2-5

ValueError: numpy.ndarray size changed, may indicate binary incompatibility. Expected 88 from C header, got 80 from PyObject

不过网上的建议是升级numpy到>=1.20.0。（ValueError: numpy.ndarray 大小已更改，可能表示二进制不兼容。从 C 头中预期为 88，从 PyObject 中得到 80）。有一个与此相关的 github 问题：https : //github.com/tensorflow/models/issues/9749。

我不确定使用什么版本来运行代码。

附录：场景#1 错误代码（numpy：1.20.0-1）

  File "model_main_tf2.py", line 113, in <module>
    tf.compat.v1.app.run()
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythonplatformapp.py", line 40, in run
    _run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef)
  File "C:Users212765830AppDataRoamingPythonPython38site-packagesabslapp.py", line 303, in run
    _run_main(main, args)
  File "C:Users212765830AppDataRoamingPythonPython38site-packagesabslapp.py", line 251, in _run_main
    sys.exit(main(argv))
  File "model_main_tf2.py", line 104, in main
    model_lib_v2.train_loop(
  File "C:Users212765830AppDataLocalContinuumanaconda3envstensorflowlibsite-packagesobject_detectionmodel_lib_v2.py", line 530, in train_loop
    train_input = strategy.experimental_distribute_datasets_from_function(
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythonutildeprecation.py", line 340, in new_func
    return func(*args, **kwargs)
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythondistributedistribute_lib.py", line 1143, in experimental_distribute_datasets_from_function
    return self.distribute_datasets_from_function(dataset_fn, options)
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythondistributedistribute_lib.py", line 1134, in distribute_datasets_from_function
    return self._extended._distribute_datasets_from_function(  # pylint: disable=protected-access
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythondistributemirrored_strategy.py", line 545, in _distribute_datasets_from_function
    return input_lib.get_distributed_datasets_from_function(
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythondistributeinput_lib.py", line 161, in get_distributed_datasets_from_function
    return DistributedDatasetsFromFunction(dataset_fn, input_workers,
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythondistributeinput_lib.py", line 1272, in __init__
    _create_datasets_from_function_with_input_context(
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythondistributeinput_lib.py", line 1936, in _create_datasets_from_function_with_input_context
    dataset = dataset_fn(ctx)
  File "C:Users212765830AppDataLocalContinuumanaconda3envstensorflowlibsite-packagesobject_detectionmodel_lib_v2.py", line 521, in train_dataset_fn
    train_input = inputs.train_input(
  File "C:Users212765830AppDataLocalContinuumanaconda3envstensorflowlibsite-packagesobject_detectioninputs.py", line 893, in train_input
    dataset = INPUT_BUILDER_UTIL_MAP['dataset_build'](
  File "C:Users212765830AppDataLocalContinuumanaconda3envstensorflowlibsite-packagesobject_detectionbuildersdataset_builder.py", line 251, in build
    dataset = dataset_map_fn(dataset, decoder.decode, batch_size,
  File "C:Users212765830AppDataLocalContinuumanaconda3envstensorflowlibsite-packagesobject_detectionbuildersdataset_builder.py", line 236, in dataset_map_fn
    dataset = dataset.map_with_legacy_function(
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythonutildeprecation.py", line 340, in new_func
    return func(*args, **kwargs)
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythondataopsdataset_ops.py", line 2679, in map_with_legacy_function
    ParallelMapDataset(
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythondataopsdataset_ops.py", line 4242, in __init__
    self._map_func = StructuredFunctionWrapper(
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythondataopsdataset_ops.py", line 3493, in __init__
    self._function.add_to_graph(ops.get_default_graph())
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythonframeworkfunction.py", line 546, in add_to_graph
    self._create_definition_if_needed()
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythonframeworkfunction.py", line 378, in _create_definition_if_needed
    self._create_definition_if_needed_impl()
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythonframeworkfunction.py", line 400, in _create_definition_if_needed_impl
    temp_graph = func_graph_from_py_func(
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythonframeworkfunction.py", line 971, in func_graph_from_py_func
    outputs = func(*func_graph.inputs)
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythondataopsdataset_ops.py", line 3485, in wrapper_fn
    ret = _wrapper_helper(*args)
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythondataopsdataset_ops.py", line 3453, in _wrapper_helper
    ret = autograph.tf_convert(func, ag_ctx)(*nested_args)
  File "C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythonautographimplapi.py", line 670, in wrapper
    raise e.ag_error_metadata.to_exception(e)
NotImplementedError: in user code:

    C:Users212765830AppDataLocalContinuumanaconda3envstensorflowlibsite-packagesobject_detectiondata_decoderstf_example_decoder.py:524 default_groundtruth_weights  *
        [tf.shape(tensor_dict[fields.InputDataFields.groundtruth_boxes])[0]],
    C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythonutildispatch.py:201 wrapper  **
        return target(*args, **kwargs)
    C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythonopsarray_ops.py:3120 ones
        output = _constant_if_small(one, shape, dtype, name)
    C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythonopsarray_ops.py:2804 _constant_if_small
        if np.prod(shape) < 1000:
    <__array_function__ internals>:5 prod

    C:Users212765830AppDataLocalContinuumanaconda3envstensorflowlibsite-packagesnumpycorefromnumeric.py:3030 prod
        return _wrapreduction(a, np.multiply, 'prod', axis, dtype, out,
    C:Users212765830AppDataLocalContinuumanaconda3envstensorflowlibsite-packagesnumpycorefromnumeric.py:87 _wrapreduction
        return ufunc.reduce(obj, axis, dtype, out, **passkwargs)
    C:Users212765830AppDataRoamingPythonPython38site-packagestensorflowpythonframeworkops.py:852 __array__
        raise NotImplementedError(
NotImplementedError: Cannot convert a symbolic Tensor (cond_2/strided_slice:0) to a numpy array. This error may indicate that you're trying to pass a Tensor to a NumPy call, which is not supported

场景#2 错误代码（numpy：1.19.2-5）

2021-02-25 09:59:47.039105: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library 'cudart64_110.dll'; dlerror: cudart64_110.dll not found
2021-02-25 09:59:47.044310: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
Traceback (most recent call last):
  File "model_main_tf2.py", line 32, in <module>
    from object_detection import model_lib_v2
  File "C:Users212765830AppDataLocalContinuumanaconda3envstensorflowlibsite-packagesobject_detectionmodel_lib_v2.py", line 29, in <module>
    from object_detection import eval_util
  File "C:Users212765830AppDataLocalContinuumanaconda3envstensorflowlibsite-packagesobject_detectioneval_util.py", line 35, in <module>
    from object_detection.metrics import coco_evaluation
  File "C:Users212765830AppDataLocalContinuumanaconda3envstensorflowlibsite-packagesobject_detectionmetricscoco_evaluation.py", line 25, in <module>
    from object_detection.metrics import coco_tools
  File "C:Users212765830AppDataLocalContinuumanaconda3envstensorflowlibsite-packagesobject_detectionmetricscoco_tools.py", line 51, in <module>
    from pycocotools import coco
  File "C:Users212765830AppDataLocalContinuumanaconda3envstensorflowlibsite-packagespycocotoolscoco.py", line 56, in <module>
    from . import mask as maskUtils
  File "C:Users212765830AppDataLocalContinuumanaconda3envstensorflowlibsite-packagespycocotoolsmask.py", line 3, in <module>
    import pycocotools._mask as _mask
  File "pycocotools_mask.pyx", line 1, in init pycocotools._mask
ValueError: numpy.ndarray size changed, may indicate binary incompatibility. Expected 88 from C header, got 80 from PyObject```

回答

def _constant_if_small(value, shape, dtype, name):
  try:
    if np.prod(shape) < 1000:
      return constant(value, shape=shape, dtype=dtype, name=name)
  except TypeError:
    # Happens when shape is a Tensor, list with Tensor elements, etc.
    pass
  return None

from tensorflow.math import reduce_prod

回答