Esempio n. 1
0
parser.add_argument('--MODEL_NAME')
parser.add_argument('--GPU')
parser.add_argument('--threshold', type=float, default=0.5)
parser.add_argument('--WEIGHT_INIT', default="Xavier")  # Truncated_Normal  or  Xavier

args = parser.parse_args()

MODEL_URL = 'http://www.vlfeat.org/matconvnet/models/beta16/imagenet-vgg-verydeep-19.mat'

os.environ["CUDA_DIVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = args.GPU

sess = tf.InteractiveSession()

image_list = utils.generate_image_mask_list_practical()

weight_init = args.WEIGHT_INIT
model_name = args.MODEL_NAME
logs_dir = './' + model_name + "/logs/"
model_dir = "./Model/"
saved_dir = './' + model_name + "/saved_model/"
threshold = args.threshold
test_saved_dir = './' + model_name + "/test_practical/" + str(threshold) + '/'

pre_threshold = False
pre_threshold_dir = test_saved_dir + 'h5_pre_threshold/'

test_batch_size = 1
test_batch_num = 85