From 21acc1059a6301bfbd1ff965976a060b893d8f02 Mon Sep 17 00:00:00 2001 From: shraddhaamohan <shraddhaa.mohan@gmail.com> Date: Sat, 9 Jan 2021 01:47:39 +0530 Subject: [PATCH] e24 1080,720 flip false 0.0015 scr thr 0.8 nms --- htc_ls_model_all.py | 6 +++--- new_eval_all.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/htc_ls_model_all.py b/htc_ls_model_all.py index 12d9225..f7f5398 100644 --- a/htc_ls_model_all.py +++ b/htc_ls_model_all.py @@ -217,9 +217,9 @@ test_cfg = dict( nms_thr=0.8, min_bbox_size=0), rcnn=dict( - score_thr=0.002, + score_thr=0.0015, nms=dict(type='nms', iou_thr=0.35), - max_per_img=50, + max_per_img=100, mask_thr_binary=0.4)) classes = [ 'water', 'pear', 'egg', 'grapes', 'butter', 'bread-white', 'jam', @@ -697,7 +697,7 @@ data = dict( dict(type='LoadImageFromFile'), dict( type='MultiScaleFlipAug', - img_scale=[(1200, 1960)], + img_scale=[(1080, 720)], flip=False, transforms=[ dict(type='Resize', keep_ratio=True), diff --git a/new_eval_all.py b/new_eval_all.py index d4c4cc1..7b2ce54 100644 --- a/new_eval_all.py +++ b/new_eval_all.py @@ -243,7 +243,7 @@ def evaluate(IMAGE_DIR, PREDICTION_FILE): if fp16_cfg is not None: wrap_fp16_model(model) - checkpoint = load_checkpoint(model,"models/epoch_23_all.pth", map_location='cpu') + checkpoint = load_checkpoint(model,"models/epoch_24_all.pth", map_location='cpu') json_file = open("data/edited_val_r3.json") -- GitLab