diff --git a/htc_ls_model_all.py b/htc_ls_model_all.py index 049d32e0d8c6951f51d426e5ba7e021b0d401929..94aa0dd2d80c67acba1e427b616a48ad50bd2895 100644 --- a/htc_ls_model_all.py +++ b/htc_ls_model_all.py @@ -214,10 +214,10 @@ test_cfg = dict( nms_pre=2500, nms_post=2000, max_num=1500, - nms_thr=0.7, + 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=100, mask_thr_binary=0.4)) @@ -403,7 +403,7 @@ test_pipeline = [ dict(type='LoadImageFromFile'), dict( type='MultiScaleFlipAug', - img_scale=[(1080,720)], + img_scale=[(850,850)], flip=False, transforms=[ dict(type='Resize', keep_ratio=True), @@ -698,7 +698,7 @@ data = dict( dict( type='MultiScaleFlipAug', img_scale=[(850, 850)], - flip=True, + flip=False, transforms=[ dict(type='Resize', keep_ratio=True), dict(type='RandomFlip'),