Skip to content
Snippets Groups Projects
Commit c6c5fec1 authored by hexiangdong2019's avatar hexiangdong2019
Browse files

debug eval_batch_scales函数有

parent 4dc00e32
Branches
No related tags found
No related merge requests found
...@@ -117,11 +117,12 @@ def eval_batch(eval_net, img_lst, crop_size=513, flip=True): ...@@ -117,11 +117,12 @@ def eval_batch(eval_net, img_lst, crop_size=513, flip=True):
def eval_batch_scales(eval_net, img_lst, scales, def eval_batch_scales(eval_net, img_lst, scales,
base_crop_size=513, flip=True): base_crop_size=513, flip=True):
print(scales)
sizes_ = [int((base_crop_size - 1) * sc) + 1 for sc in scales] sizes_ = [int((base_crop_size - 1) * sc) + 1 for sc in scales]
probs_lst = eval_batch(args, eval_net, img_lst, crop_size=sizes_[0], flip=flip) probs_lst = eval_batch(eval_net, img_lst, crop_size=sizes_[0], flip=flip)
print(sizes_) print(sizes_)
for crop_size_ in sizes_[1:]: for crop_size_ in sizes_[1:]:
probs_lst_tmp = eval_batch(args, eval_net, img_lst, crop_size=crop_size_, flip=flip) probs_lst_tmp = eval_batch(eval_net, img_lst, crop_size=crop_size_, flip=flip)
for pl, _ in enumerate(probs_lst): for pl, _ in enumerate(probs_lst):
probs_lst[pl] += probs_lst_tmp[pl] probs_lst[pl] += probs_lst_tmp[pl]
...@@ -145,7 +146,6 @@ def net_eval(): ...@@ -145,7 +146,6 @@ def net_eval():
bi = 0 bi = 0
image_num = 0 image_num = 0
for i, line in enumerate(img_lst): for i, line in enumerate(img_lst):
print(line)
img_path = os.path.join('./datasets/JPEGImages/', line+'.bmp') img_path = os.path.join('./datasets/JPEGImages/', line+'.bmp')
msk_path = os.path.join('./datasets/SegmentationClass/', line+'.png') msk_path = os.path.join('./datasets/SegmentationClass/', line+'.png')
img_ = cv2.imread(img_path) img_ = cv2.imread(img_path)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment