tensorflow / tensorflow/models

Error in object detection using ssd_mobilenet_v1_ppn_

Open
#8,329 0 comments 0 reactions 1 assignee View on GitHub

@sguada is already working on this.

Since May 11, 2020.

models:research type:bug
Dominant language
Python
Stars
77.7k
Forks
44.8k
PR merge metrics
No merged PRs in 30d

Description

System information

  • Have I written custom code (as opposed to using a stock example script provided in TensorFlow): No
  • OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Win10
  • Mobile device (e.g., Pixel 4, Samsung Galaxy 10) if the issue happens on mobile device: no
  • TensorFlow installed from (source or binary): binary
  • TensorFlow version (use command below): 1.15.0
  • Python version: 3.7
  • Bazel version (if compiling from source): na
  • GCC/Compiler version (if compiling from source): na
  • CUDA/cuDNN version: 10.0 / 7.6
  • GPU model and memory:

When i train per detection using pretrained model here:
ssd_mobilenet_v1_ppn_shared_box_predictor_300x300_coco14_sync_2018_07_03

I have follow error:
Traceback (most recent call last):
File "object_detection/model_main.py", line 109, in
tf.app.run()
File "C:\Users\enrico.beltramo.conda\envs\objdecbase\lib\site-packages\tensorflow_core\python\platform\app.py", line 40, in run
_run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef)
File "C:\Users\enrico.beltramo.conda\envs\objdecbase\lib\site-packages\absl\app.py", line 299, in run
_run_main(main, args)
File "C:\Users\enrico.beltramo.conda\envs\objdecbase\lib\site-packages\absl\app.py", line 250, in _run_main
sys.exit(main(argv))
File "object_detection/model_main.py", line 105, in main
tf.estimator.train_and_evaluate(estimator, train_spec, eval_specs[0])
File "C:\Users\enrico.beltramo.conda\envs\objdecbase\lib\site-packages\tensorflow_estimator\python\estimator\training.py", line 473, in train_and_evaluate
return executor.run()
File "C:\Users\enrico.beltramo.conda\envs\objdecbase\lib\site-packages\tensorflow_estimator\python\estimator\training.py", line 613, in run
return self.run_local()
File "C:\Users\enrico.beltramo.conda\envs\objdecbase\lib\site-packages\tensorflow_estimator\python\estimator\training.py", line 714, in run_local
saving_listeners=saving_listeners)
File "C:\Users\enrico.beltramo.conda\envs\objdecbase\lib\site-packages\tensorflow_estimator\python\estimator\estimator.py", line 370, in train
loss = self._train_model(input_fn, hooks, saving_listeners)
File "C:\Users\enrico.beltramo.conda\envs\objdecbase\lib\site-packages\tensorflow_estimator\python\estimator\estimator.py", line 1161, in _train_model
return self._train_model_default(input_fn, hooks, saving_listeners)
File "C:\Users\enrico.beltramo.conda\envs\objdecbase\lib\site-packages\tensorflow_estimator\python\estimator\estimator.py", line 1191, in _train_model_default
features, labels, ModeKeys.TRAIN, self.config)
File "C:\Users\enrico.beltramo.conda\envs\objdecbase\lib\site-packages\tensorflow_estimator\python\estimator\estimator.py", line 1149, in _call_model_fn
model_fn_results = self._model_fn(features=features, **kwargs)
File "C:\Projects\CORAL\ObjectDetection\models-master\research\object_detection\model_lib.py", line 308, in model_fn
features[fields.InputDataFields.true_image_shape])
File "C:\Projects\CORAL\ObjectDetection\models-master\research\object_detection\meta_architectures\ssd_meta_arch.py", line 600, in predict
preprocessed_inputs)
File "C:\Projects\CORAL\ObjectDetection\models-master\research\object_detection\models\ssd_mobilenet_v1_ppn_feature_extractor.py", line 83, in extract_features
'image_features': image_features['Conv2d_11_pointwise']
File "C:\Projects\CORAL\ObjectDetection\models-master\research\object_detection\models\feature_map_generators.py", line 781, in pooling_pyramid_feature_maps
image_features = image_features[image_features.keys()[0]]
TypeError: 'dict_keys' object is not subscriptable

with other pretrained models (i.e mobile net v1), it works correctly

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.