search

avolkov1 / keras_experiments

Experimental Keras libraries and examples.
The Unlicense
86 stars 16 forks source link

ImportError: No module named keras_frcnn #4

Closed bzamecnik closed 7 years ago

bzamecnik commented 7 years ago 
$ PYTHONPATH=. CUDA_VISIBLE_DEVICES=3 python examples/faster_rcnn/train_frcnn_mgpu.py --epochs=4
Using TensorFlow backend.
Traceback (most recent call last):
  File "examples/faster_rcnn/train_frcnn_mgpu.py", line 21, in <module>
    from keras_frcnn import config, data_generators
ImportError: No module named keras_frcnn

It implicitly assumes that https://github.com/yhenon/keras-frcnn is installed and available for import.

avolkov1 commented 7 years ago

@bzamecnik Yes, exactly. I'm not going to copy all the code from keras-frcnn. At the top of train_frcnn_mgpu.py I commented:

# Based on: https://github.com/yhenon/keras-frcnn/blob/master/train_frcnn.py
# Clone repo: https://github.com/yhenon/keras-frcnn
# add this example into the repo with keras_exp in the PYTHONPATH and run.
# python train_frcnn_mgpu.py -p /path/to/pascalvoc

I added this example for illustration with issue #3. The reporter there wanted to run train_frcnn.py with multiple GPUs. That example is not so simple run with multiple GPUs. I fixed a bug dealing with multiple outputs not being handled correctly, but the other issues have nothing to do with multigpu logic code. Refer to my explanation here: explanation.

bzamecnik commented 7 years ago

OK. Thanks for info.