EdjeElectronics / TensorFlow-Object-Detection-API-Tutorial-Train-Multiple-Objects-Windows-10

How to train a TensorFlow Object Detection Classifier for multiple object detection on Windows
Apache License 2.0
2.92k stars 1.3k forks source link

Unabe to train in TF2.0 #412

Open suravijayjilla opened 4 years ago

suravijayjilla commented 4 years ago

Hi,

i tried to run this code in tensorflow-gpu, but i am facing this issue while execute train command,

(tensor_gpu) C:\Users\admin\Downloads\models\research\object_detection>python train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/faster_rcnn_inception_v2_pets.config 2019-12-02 16:21:53.124045: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_100.dll Traceback (most recent call last): File "train.py", line 48, in from tensorflow.contrib import framework as contrib_framework ModuleNotFoundError: No module named 'tensorflow.contrib'

Please give some suggestions to solve this issue...

WaitNothing commented 4 years ago

look at issue #383