Arta Asadi

@wangxinyue11 yes I resized images to 320*320 using pytorch transforms.

Thanks, it is really needed. I found your work really interesting and I am using it in my work. So, Thanks!

@doradocrux I have executed the code for prediction, unfortunately the time of the training was very inefficient and high, so I deleted the code. Also, I did not have so...