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