From e274d9499771dd9d61371765dbe77f7411b6597d Mon Sep 17 00:00:00 2001
From: RohitMidha23 <rohit.midha23@gmail.com>
Date: Mon, 14 Dec 2020 15:20:27 +0530
Subject: [PATCH] 850 850 flip False scr0.0015 nms0.8
---
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 049d32e..94aa0dd 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'),
--
GitLab