Skip to content

legokichi/keras-pix2pix

Repository files navigation

keras-pix2pix keras-u-net

working in progress

ImageGAN only.

ref

install

git clone --recursive https://github.com/legokichi/keras-pix2pix.git
pyenv shell anaconda3-4.1.1
sudo apt-get install graphviz
conda install theano pygpu
pip install tensorflow-gpu
pip install keras
pip install mypy
pip install pydot_ng

type check

mypy --ignore-missing-imports train.py 

train

env CUDA_VISIBLE_DEVICES=0 python train.py --lr=0.0001 --dice_coef
env CUDA_VISIBLE_DEVICES=1 python train.py --lr=0.0001 --dice_coef --data_aug

model

unet

disc

U-net Web Server

setup

pip install gunicorn flask

on gnicorn

gunicorn -w 4  -b 0.0.0.0:8888  server:app

on flask

env FLASK_APP=server.py flask run --host=0.0.0.0 --port 8888

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published