need trained model,thanks!

[Kratosyxr]

Thank you for your code.i am very interested this .i have run this code successfully.but my model test is not well.and my gpu is 1070 .it is just 8g.so my model is not good enough.can you upload your trained model.i would be very grateful if you could upload your model.