Open ajinkya933 opened 5 years ago
i have the same problem, too. Training works on icdar2015 dataset, But doesn't work on my dataset. Not even a box detected after testing on a model at step 300.
Did You manage to solve it?
@ajinkya933 @mohamad-hoseini Were you guys able to resolve this issue?
I followed your code but when I retrain the model using python2 I coudnt reproduce your results, rather all the output files donot have any box coordinates,
Training batchsize=16 Platform: AWS p2xlarge (1 Nvidia GPU) python version: python2
I followed a discusion thread in issues which said "pixel_link-mingyu.guo" implementation is much better than yours and he has done converting your code to python3.
But in this case even that failed, heres his repo https://github.com/baby313/pixel_link
How can I get output by retraining?
File sizes of trained model are as follows: