thomasjhuang / deep-learning-for-document-dewarping Goto Github PK

Hi, can you please add scripts for generating a synthetic training dataset.

Thanks for this repo!
I have some questions about the results of different methods.
(1) Can pix2pixHD method gets a better result than DocUNet ?
(2) Have you changed some things about the original pix2pixHD method?

Good day!;)
I try to train model on kaggle dataset, i use a train options from readme ... but get a error ... =(
create web directory ./checkpoints/kaggle/web...
Traceback (most recent call last):
File "train.py", line 71, in
Variable(data['image']), Variable(data['feat']), infer=save_fake)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/parallel/data_parallel.py", line 153, in forward
return self.module(*inputs[0], **kwargs[0])
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "/content/drive/Shared drives/UNLIMITED/ducunet/deep-learning-for-document-dewarping-master/models/pix2pixHD_model.py", line 163, in forward
fake_image = self.netG.forward(input_concat)
File "/content/drive/Shared drives/UNLIMITED/ducunet/deep-learning-for-document-dewarping-master/models/networks.py", line 180, in forward
output_prev = model_upsample(model_downsample(input_i) + output_prev)
RuntimeError: The size of tensor a (398) must match the size of tensor b (400) at non-singleton dimension 2

How i can solve it?;)

Hi! I have a bit of a struggle trying to reproduce training procedure. As I am aware, data folder is in gitignore file. I've tried to take the same folder from official pix2pix repository that you provided, but it didn't work. May be this is not due to data folder, but anyway help is appreciated

I use the script to generate some distorted handwritten Chinese character data sets, and then modify the size of training and original image, but the effect is not good. Does the principle of this code apply to handwritten Chinese character data sets? If so, do you need to modify some parameters?

Hi, thanks for sharing this great project.
Can you share a pre-trained model so we can evaluate easily?

Excuse me, when I run test.py, it reported "./checkpoints/kaggle/latest_net_G.pth not exists yet!".

excuse me, when i try to train this model, i meet a problem, can you help me to solve it?
The problem: "RuntimeError: The size of tensor a (245) must match the size of tensor b (256) at non-singleton dimension 2"
thank you!!!