From a0301f7cb4f0cd894929556e39c595b4cf583f2c Mon Sep 17 00:00:00 2001 From: shraddhaamohan <shraddhaa.mohan@gmail.com> Date: Fri, 8 Jan 2021 19:28:28 +0530 Subject: [PATCH] 1200,1960 flip false 0.002 scr thres nms 0.8 max dets 50 --- htc_ls_model_all.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/htc_ls_model_all.py b/htc_ls_model_all.py index 2414ff6..12d9225 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.0015, + score_thr=0.002, nms=dict(type='nms', iou_thr=0.35), - max_per_img=100, + max_per_img=50, mask_thr_binary=0.4)) classes = [ 'water', 'pear', 'egg', 'grapes', 'butter', 'bread-white', 'jam', @@ -697,8 +697,8 @@ data = dict( dict(type='LoadImageFromFile'), dict( type='MultiScaleFlipAug', - img_scale=[(1080, 720)], - flip=True, + img_scale=[(1200, 1960)], + flip=False, transforms=[ dict(type='Resize', keep_ratio=True), dict(type='RandomFlip'), -- GitLab