fcn网络8s复现
FCN.py import numpy as np import tensorflow as tf import scipy . io as scio from scipy import misc import sys import logging import datetime FLAGS = tf . flags . FLAGS tf . flags . DEFINE_integer ( 'batchsize' , '10' , 'trainning batchsize' ) #flag tf . flags . DEFINE_float ( 'learning_rate' , '1e-4' , 'learning_rate' ) #flag tf . flags . DEFINE_bool ( 'reuse' , "False" , "reuse the pretrained model" ) tf . flags . DEFINE_bool ( 'train' , "True" , "train or test" ) tf . flags . DEFINE_string ( 'checkpoint' , "checkpoint" , "dir to save model" ) tf . flags . DEFINE_string ( 'log' , "log" , "dir