提交 36d21984 编写于 作者: L LielinJiang 提交者: qingqing01

fix batch generate bug in eval.py (#2482)

* fix batch generate bug in eval.py.
上级 ee5fdb92
......@@ -112,7 +112,7 @@ else:
batches = dataset.get_batch_generator(batch_size, total_step)
if args.use_py_reader:
py_reader.decorate_tensor_provider(lambda :[ (yield b[1],b[2]) for b in batches])
py_reader.decorate_tensor_provider(lambda :[ (yield b[0],b[1]) for b in batches])
py_reader.start()
sum_iou = 0
......@@ -137,3 +137,5 @@ for i in range(total_step):
mp = (wrong + right) != 0
miou2 = np.mean((right[mp] * 1.0 / (right[mp] + wrong[mp])))
print('step: %s, mIoU: %s' % (i + 1, miou2))
print('eval done!')
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册