h

app.py

@@ -43,7 +43,20 @@ else:
   init_detector = sys.modules['init_detector']
 
 ########## utils
-
+def draw_conts(im,conts,fcolor,stroke):
+  opacity = .3
+  im =im.copy()
+  # fcolor =fcolor if fcolor else get_random_color()
+  frontmm = np.full_like(im,fcolor,np.uint8)
+  mask = np.zeros_like(im)
+  for cont in conts:
+    cv2.drawContours(mask, [cont], -1, color=[int(opacity*255)]*3, thickness=cv2.FILLED)
+  mask = cv2.cvtColor(mask, cv2.COLOR_BGR2GRAY)
+  rgba = np.dstack((frontmm, mask))
+  im = utilss.overlay_image_alpha(im,rgba)
+  # drawline
+  cv2.drawContours(im, conts, -1, color=fcolor, thickness=stroke)
+  return im
 def image_resize(image, width = None, height = None, inter = cv2.INTER_AREA):
     dim = None
     (h, w) = image.shape[:2]
@@ -89,7 +102,8 @@ def loading_model():
   from mmdet.apis import set_random_seed
   set_random_seed(0, deterministic=False)
   cfg = Config.fromfile('./cfgsn.py')
-  checkpoint = "./200s.pth"
+  # checkpoint = "./200s.pth"
+  checkpoint = "./epoch_100.pth"
   model1 = init_detector(cfg, checkpoint, device="cpu")
   if model1.cfg.data.test['type'] == 'ConcatDataset':
     model1.cfg.data.test.pipeline = model1.cfg.data.test['datasets'][0].pipeline
@@ -114,21 +128,22 @@ def main():
 
   st.sidebar.write(" ------ ")
   st.sidebar.info('Options')
-  fcolor = st.sidebar.selectbox('boundary color', ['red','green','blue'])
   extra_postprocess = st.sidebar.checkbox('perform extra postprocessing',value=True)
+  fcolor = st.sidebar.selectbox('boundary color', ['red','green','blue'],index =0)
+  fstroke = st.sidebar.selectbox('boundary thickness', ['1','2','3'] ,index=1 )
   # do_thresh_box = st.sidebar.checkbox('adaptive threshold')
 
   if btnpredicrupload and f is not None:
     tfile = tempfile.NamedTemporaryFile(delete=True)
     tfile.write(f.read())
     imgpath=tfile.name
-    run_app(imgpath ,fcolor,extra_post=extra_postprocess)
+    run_app(imgpath ,fcolor,extra_post=extra_postprocess,fstroke=fstroke)
   if submitted:
     st.empty()
     directory ='./imgs/'
     pic = os.path.join(directory, option)
     imgpath=directory+option
-    run_app(imgpath ,fcolor,extra_post=extra_postprocess)
+    run_app(imgpath ,fcolor,extra_post=extra_postprocess,fstroke=fstroke)
   # inp = st.text_input('t2','')
   # if st.button('run'):
   #   subprocess.run(inp.split(' '))
@@ -165,13 +180,17 @@ def run_app(imgpath,fcolor,**kwargs):
   ff= fixer(img,preds,img_metas,downsample_ratio)
   polys= ff.polysfin3
   if kwargs['extra_post']:
-    polys =find_polys4(ff) 
+    try:
+      polys =find_polys4(ff) 
+    except:
+      print('error polys4')
+      polys= ff.polysfin3
   
 
   # with open('./0.json','r') as f:
   #   res = json.load(f)
   ################
-  mm = utilss.draw_conts(img,[poly2cont(p) for p in polys],fcolor)
+  mm = draw_conts(img,[poly2cont(p) for p in polys],fcolor,stroke=int(kwargs['fstroke']))
   ## mm=image_resize(mm, width = 640)
   #######
   # mm=preds[1]*preds[0]