Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

COCO-trained models for tf2,where can download? #8752

Closed
xpdrycry opened this issue Jun 30, 2020 · 2 comments
Closed

COCO-trained models for tf2,where can download? #8752

xpdrycry opened this issue Jun 30, 2020 · 2 comments
Assignees
Labels
models:research models that come under research directory type:bug Bug in the code

Comments

@xpdrycry
Copy link

Prerequisites

Please answer the following questions for yourself before submitting an issue.

  • [ X] I am using the latest TensorFlow Model Garden release and TensorFlow 2.

1. The entire URL of the file you are using

https://github.com/tensorflow/models/tree/master/research/...

2. Describe the bug

python3 legacy/train.py --logtostderr --train_dir=voc --pipeline_config_path=voc_lib/faster_rcnn_resnet101_pets.config

ValueError: faster_rcnn_resnet101 is not supported. See model_builder.py for features extractors compatible with different versions of Tensorflow

3. Steps to reproduce

WARNING:tensorflow:From /home/c/.local/lib/python3.6/site-packages/absl/app.py:250: main (from main) is deprecated and will be removed in a future version.
Instructions for updating:
Use object_detection/model_main.py.
W0630 14:38:04.062839 140272845739840 deprecation.py:323] From /home/c/.local/lib/python3.6/site-packages/absl/app.py:250: main (from main) is deprecated and will be removed in a future version.
Instructions for updating:
Use object_detection/model_main.py.
Traceback (most recent call last):
File "legacy/train.py", line 186, in
tf.app.run()
File "/home/c/.local/lib/python3.6/site-packages/tensorflow/python/platform/app.py", line 40, in run
_run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef)
File "/home/c/.local/lib/python3.6/site-packages/absl/app.py", line 299, in run
_run_main(main, args)
File "/home/c/.local/lib/python3.6/site-packages/absl/app.py", line 250, in _run_main
sys.exit(main(argv))
File "/home/c/.local/lib/python3.6/site-packages/tensorflow/python/util/deprecation.py", line 324, in new_func
return func(*args, **kwargs)
File "legacy/train.py", line 182, in main
graph_hook_fn=graph_rewriter_fn)
File "/usr/local/lib/python3.6/dist-packages/object_detection-0.1-py3.6.egg/object_detection/legacy/trainer.py", line 248, in train
detection_model = create_model_fn()
File "/usr/local/lib/python3.6/dist-packages/object_detection-0.1-py3.6.egg/object_detection/builders/model_builder.py", line 950, in build
add_summaries)
File "/usr/local/lib/python3.6/dist-packages/object_detection-0.1-py3.6.egg/object_detection/builders/model_builder.py", line 510, in _build_faster_rcnn_model
_check_feature_extractor_exists(frcnn_config.feature_extractor.type)
File "/usr/local/lib/python3.6/dist-packages/object_detection-0.1-py3.6.egg/object_detection/builders/model_builder.py", line 208, in _check_feature_extractor_exists
'Tensorflow'.format(feature_extractor_type))

4. Expected behavior

5. Additional context

6. System information

import tensorflow as tf
print(tf.version.GIT_VERSION, tf.version.VERSION)
v2.2.0-rc4-8-g2b96f3662b 2.2.0

##according to model_builder.py, where is pretraining model for tf2?

@xpdrycry xpdrycry added models:research models that come under research directory type:bug Bug in the code labels Jun 30, 2020
@jch1
Copy link
Member

jch1 commented Jun 30, 2020

We currently only support TF1 in the Object Detection API --- so this is why things are breaking if you try to use the models with a TF2 installation. But we plan to make TF2 models available very soon (in the next couple of weeks), so please stay tuned.

@pkulzc
Copy link
Contributor

pkulzc commented Jul 14, 2020

We just released Object Detection API for TF2. Please see our updated doc. Please note that for TF2 training, train.py is no longer supported.

@pkulzc pkulzc closed this as completed Jul 14, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
models:research models that come under research directory type:bug Bug in the code
Projects
None yet
Development

No branches or pull requests

4 participants