InvalidArgumentError

I got the following error while running "python train_frcnn.py -o simple -p train.txt "

Using TensorFlow backend.
Parsing annotation files
Training images per class:
{'bg': 0, 'table': 418}
Num classes (including bg) = 2
Config has been written to config.pickle, and can be loaded when testing to ensure correct results
Num train samples 287
Num val samples 51
Traceback (most recent call last):
  File "/Users/ketulshah/anaconda3/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1628, in _create_c_op
    c_op = c_api.TF_FinishOperation(op_desc)
tensorflow.python.framework.errors_impl.InvalidArgumentError: Shape must be rank 1 but is rank 0 for 'bn_conv1/Reshape_4' (op: 'Reshape') with input shapes: [1,1,1,64], [].

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "train_frcnn.py", line 122, in <module>
    shared_layers = nn.nn_base(img_input, trainable=True)
  File "/Users/ketulshah/table-detection/keras-frcnn/keras_frcnn/resnet.py", line 180, in nn_base
    x = FixedBatchNormalization(axis=bn_axis, name='bn_conv1')(x)
  File "/Users/ketulshah/anaconda3/lib/python3.6/site-packages/keras/engine/base_layer.py", line 457, in __call__
    output = self.call(inputs, **kwargs)
  File "/Users/ketulshah/table-detection/keras-frcnn/keras_frcnn/FixedBatchNormalization.py", line 73, in call
    epsilon=self.epsilon)
  File "/Users/ketulshah/anaconda3/lib/python3.6/site-packages/keras/backend/tensorflow_backend.py", line 1908, in batch_normalization
    mean = tf.reshape(mean, (-1))
  File "/Users/ketulshah/anaconda3/lib/python3.6/site-packages/tensorflow/python/ops/gen_array_ops.py", line 6482, in reshape
    "Reshape", tensor=tensor, shape=shape, name=name)
  File "/Users/ketulshah/anaconda3/lib/python3.6/site-packages/tensorflow/python/framework/op_def_library.py", line 787, in _apply_op_helper
    op_def=op_def)
  File "/Users/ketulshah/anaconda3/lib/python3.6/site-packages/tensorflow/python/util/deprecation.py", line 488, in new_func
    return func(*args, **kwargs)
  File "/Users/ketulshah/anaconda3/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 3274, in create_op
    op_def=op_def)
  File "/Users/ketulshah/anaconda3/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1792, in __init__
    control_input_ops)
  File "/Users/ketulshah/anaconda3/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1631, in _create_c_op
    raise ValueError(str(e))
ValueError: Shape must be rank 1 but is rank 0 for 'bn_conv1/Reshape_4' (op: 'Reshape') with input shapes: [1,1,1,64], [].

[ketul93]

With keras 2.0.3 I am ubable to use GPU …

How to utilise GPU (NVIDIA TITAN Xp) with keras 2.0.3 ?

I read that GPU is only supported for Keras >= 2.1.1

[Ersho]

As it says in requirements.txt, you need to install keras 2.0.3.

You can simply run command

pip install keras==2.0.3