[Fix] Change SBC case to DBC case (#632)

This commit is contained in:
Hongbin Sun 2021-11-30 15:17:07 +08:00 committed by GitHub
parent 1c8ab80409
commit d1d2c1f469
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -125,6 +125,8 @@ def eval_hmean(results,
best_result = dict(hmean=-1)
for iter in range(3, 10):
thr = iter * 0.1
if thr < score_thr:
continue
top_preds = select_top_boundary(preds, pred_scores, thr)
if metric == 'hmean-iou':
result, img_result = hmean_iou.eval_hmean_iou(
@ -138,9 +140,9 @@ def eval_hmean(results,
output_ranklist(img_result, img_infos, rank_list)
print_log(
'thr {0:.1f}, recall{1[recall]:.3f}, '
'thr {0:.2f}, recall: {1[recall]:.3f}, '
'precision: {1[precision]:.3f}, '
'hmean:{1[hmean]:.3f}'.format(thr, result),
'hmean: {1[hmean]:.3f}'.format(thr, result),
logger=logger)
if result['hmean'] > best_result['hmean']:
best_result = result