Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[ASK] about the job of document cleanup #658

Open
shanyun123456 opened this issue Oct 12, 2021 · 9 comments
Open

[ASK] about the job of document cleanup #658

shanyun123456 opened this issue Oct 12, 2021 · 9 comments

Comments

@shanyun123456
Copy link

shanyun123456 commented Oct 12, 2021

Is there the pre-trained model of the job of document cleanup,Would you mind uploading the pre-trained model?
@soumyadeepdey

https://paperswithcode.com/paper/high-resolution-image-synthesis-and-semantic
You have uploaded the implementation of torch in this url,but actually it's the implementation of tf.
if you have the implementation of troch,would you mind uploading it?

@shanyun123456
Copy link
Author

@soumyadeepdey

@soumyadeepdey
Copy link
Contributor

soumyadeepdey commented Oct 13, 2021 via email

@shanyun123456
Copy link
Author

Sorry about that,i have uploaded a other url.
The right url is https://paperswithcode.com/paper/light-weight-document-image-cleanup-using

@shanyun123456
Copy link
Author

Would mind uploading the pre-treained model of M32-color?

@soumyadeepdey
Copy link
Contributor

soumyadeepdey commented Oct 13, 2021 via email

@shanyun123456
Copy link
Author

How about the pre-trained model of M32-gary?
I have tested the model in github, but this problem occered,Would mind checking that?

infer code:
#Input image
test_img_name = 'dataset/sample_data/image_56.png'
##########################################################################################
#calling inference engine
out_img_name = 'test_out1.jpeg'
infer_image('checkpoints/M16_noisyoffice.json','checkpoints/M16_noisyoffice_epoch-910.hdf5',test_img_name,out_img_name)

the problem:
NameError: name 'convert2gray' is not defined

thanks

@shanyun123456
Copy link
Author

shanyun123456 commented Oct 13, 2021

Hi,When i use your code train new model in gpu,it seems always occer the problem of OOM.
How can i change your code to fix that problem?
I set train_batch_size is 1

the problem is:
.....
six.raise_from(core._status_to_exception(e.code, message), None)
File "", line 3, in raise_from
tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[512,512,3,3] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc [Op:Conv2DBackpropFilter]
2021-10-13 15:53:55.915987: W tensorflow/core/kernels/data/generator_dataset_op.cc:103] Error occurred when finalizing GeneratorDataset iterator: Failed precondition: Python interpreter state is not initialized. The process may be terminated.

@sususu1
Copy link

sususu1 commented Dec 31, 2021

hi, do u know how to generate gt data

@soumyadeepdey
Copy link
Contributor

soumyadeepdey commented Dec 31, 2021 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants