diff --git a/htc_ls_model_all.py b/htc_ls_model_all.py index c1bc0e20213c2b2c61481368d93b2bd8a7b8a967..706e7cb8ae9b74c81c8a7f43c69f23c67be910b6 100644 --- a/htc_ls_model_all.py +++ b/htc_ls_model_all.py @@ -220,7 +220,7 @@ test_cfg = dict( score_thr=0.003, nms=dict(type='nms', iou_thr=0.35), max_per_img=100, - mask_thr_binary=0.4)) + mask_thr_binary=0.45)) classes = [ 'water', 'pear', 'egg', 'grapes', 'butter', 'bread-white', 'jam', 'bread-whole-wheat', 'apple', 'tea-green', 'white-coffee-with-caffeine', @@ -404,7 +404,7 @@ test_pipeline = [ dict( type='MultiScaleFlipAug', img_scale=[(1080,720)], - flip=False, + flip=True, transforms=[ dict(type='Resize', keep_ratio=True), dict(type='RandomFlip'),