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