Hi @Dawars ,
You have done a great job on implementing your version of DocMAE. I have do some test using your u2netp.pth pretrained model. I‘m interesting how you train the u2netp model, especially using which dataset and the network parameters. Could you please share these infomation. Thank you!