kgpml / deep-vessel Goto Github PK

Optimize the preprocessor to reduce the time wasted by rejecting non-valid random points

h_pool2_flat = tf.reshape(h_pool2, [-1, 7764])
why you set 7764?

Dimension size must be evenly divisible by 3136 but is 36864 for 'h_fc1/Reshape' (op: 'Reshape') with input shapes: [64,3,3,64], [2] and with input tensors computed as partial shapes: input[1] = [?,3136].

if name == "main":
sys.argv = ['Test.py','--fchu1','128',
'--format','png',
'--out','../../Data/DRIVE/tmp/',
'--inp','../../Data/DRIVE/test/',
'--model','../../Data/models/model1/model.ckpt-499']
main()

New FCHU1 = 128
New OUT_DIR = ../../Data/DRIVE/tmp/
New IN_DIR = ../../Data/DRIVE/test/
New MODEL_PATH = ../../Data/models/model1/model.ckpt-499
New FORMAT = png
INFO:tensorflow:Restoring parameters from ../../Data/models/model1/model.ckpt-499
Working on image 1
C:\ProgramData\Anaconda3\lib\site-packages\skimage\external\tifffile\tifffile.py:2611: RuntimeWarning: py_decodelzw encountered unexpected end of stream
strip = decompress(strip)

ValueError Traceback (most recent call last)
in ()
5 '--inp','../../Data/DRIVE/test/',
6 '--model','../../Data/models/model1/model.ckpt-499']
----> 7 main()

in main()
13 os.makedirs(OUT_DIR)
14
---> 15 decode(test, mask_test)

in decode(test, mask_test)
72 #print " Feed shape = (%d, %d)" % feed.shape
73 chunk = np.array(list(map(lambda p: nbd(image, p), row_col[i:i_next, j:j_next].reshape(-1,2))))
---> 74 feed[:chunk.size] = chunk
75
76 # Subtract training mean image

ValueError: could not broadcast input array from shape (40,2883) into shape (64,2883)