chris10m / rfb-text-detection Goto Github PK

@Chris10M

• Can you add the training script.
• Can you add prediction using GPU script.
• Can this code detect entire text-lines, or is it limited to words?
• Can you add the option to extract the detected text?

Waiting for your reply

So I downloaded the model from the google drive link mentioned in the readme. On running the run_demo.py code I get the following error:

Traceback (most recent call last):
File "run_demo.py", line 113, in
rfb_text_model = RFBText(model_weights_path=model_path)()
File "/media/sf_ubuntu_shared/text_detection/RFB-Text-Detection/model.py", line 153, in call
model = self.build()
File "/media/sf_ubuntu_shared/text_detection/RFB-Text-Detection/model.py", line 112, in build
C5 = self.backbone.get_layer('bn5c_branch2c').output
File "/home/pratik/.virtualenvs/tf_2.0-keras/lib/python3.8/site-packages/tensorflow/python/keras/engine/training.py", line 2398, in get_layer
raise ValueError('No such layer: ' + name + '.')
ValueError: No such layer: bn5c_branch2c.

Any help would be great. Thanks.

The Drive link which provide for downloading pretrained weights is not working
https://drive.google.com/open?id=1mw8v_VV1KidyrqY_0A_oSYxRhPex4oKY

Hello,
I am trying to run run_demo.py in windows system, i am getting following error,

Hello,
I am trying to run run_demo.py in windows system, i am getting following error,

@Chris10M could you please upload the training code

@Chris10M training code

Hi Chris, can you upload the trained model?