Hi, Im trying to finetuning the model with my custom data. In the source code train.py , It uses dataset/train/data1024_greyV2/color as default directory of training set. In this directory, there is some OOO.gw files.

What is gw file and how I can make image file as gw file?

Hello！ I used your Distorted Image code to generate my own training dataset, and I trained the model using this dataset. But it seems that the output model is not able to correct the horizontal orientation of the image, like the picture below shows. You can see that the left side and the right side of the flated image is also as same as the distort image, keep tilt. But on the vertical direction, like the up side and bottom side have been corrected. The same problem happens on the all the test pictures. SO I check the output regress of the model, the displacement value of horizontal direction is very small, the max absolute value is only around 0.1. But the displacement value of vertical direction is relatively big, which seems reasonable. For detecting this problem, I have attempted to check the loss function, but it seems correct. And also I checked the generated label_regress which contains the flow of x and y. I just used the label_regess to flat the distorted image, then the distorted image can be flated in a totally correct way. So the generated training dataset is correct. Do you have any idea what the problem might be? Thanks for your help in advance.

With some fixes I could run a test and get result from 130 dataset(you provided) and my own pictures. Now I'm trying to train the model with some more pictures and I can't find a way. Can you provide a train source code and sample dataset?

Hi, Thanks for sharing this excellent work for document rectification! But I can't run through the released codes with data structure like you arranged.Please help me!