search

JianqiangRen / AAMS

Official repository of the paper 'Attention-aware Multi-stroke Style Transfer' in CVPR 2019
https://sites.google.com/view/yuanyao/attention-aware-multi-stroke-style-transfer
MIT License
79 stars 14 forks source link

Program is too slow #10

Closed adaxidedakaonang closed 2 years ago

adaxidedakaonang commented 2 years ago

I run your program with RTX 2080Ti and the time consumption is about 500 S. The content size is (800,800,3), style size is (512,512,3), and the attention size is (1,100,100,512).

adaxidedakaonang commented 2 years ago

The tested time is about code "sess.run()" in test.py

JianqiangRen commented 2 years ago

Maybe there is some problem with your environment? I have never met this before, even with only CPU, 500s for one images is inconceivable

adaxidedakaonang commented 2 years ago

New test result is : a image with size of 512X512 cost 1.2S。在进行1024图像比较时，程序会改变图像大小至800？，运行时间为1.5S