Skip to content

Commit

Permalink
check dataset path.
Browse files Browse the repository at this point in the history
  • Loading branch information
dwSun authored and dwSun committed Feb 1, 2018
1 parent 7d3bcfb commit a345f3d
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
import time

import tensorflow as tf
import os

from utils import (_B_MEAN, _G_MEAN, _R_MEAN, _mean_image_subtraction)

Expand Down Expand Up @@ -33,6 +34,7 @@ def read_and_decode(filename_queue):


def inputs(data_set, train=True, batch_size=1, num_epochs=1, upsample_factor_for_whole_net=32):
assert os.path.exists(data_set), '[{0}] not exist!!!'.format(data_set)
if not num_epochs:
num_epochs = None

Expand Down

0 comments on commit a345f3d

Please sign in to comment.