- System
- Ubuntu 16.04 or later
- NVIDIA GPU + CUDA 9.2(later version also can be sucess)
- python 3.7
- install require Packages
sh install.sh
- rename a figure:
mv heat/bg.jpg heat/.jpg
- prepare model file
Please download the pre-trained model[Baidu(Password:wiu9)] and put those under 'Param'.
- prepare sketch image file for post in current path
size of sketch image is 512X512
server:
python3.7 inference.py
client:
python3.7 request.py