Update app.py

remove crf operation

app.py

@@ -12,9 +12,9 @@ os.system("pip3 install collections")
 os.system("pip3 install torchvision")
 os.system("pip3 install einops")
 aaaa=0
-os.system("pip3 install pydensecrf")
+#os.system("pip3 install pydensecrf")
 #os.system("pip install argparse")
-import pydensecrf.densecrf as dcrf
+#import pydensecrf.densecrf as dcrf
 from PIL import Image
 import torch
 import torch.nn.functional as F
@@ -95,7 +95,7 @@ def test(gpu_id, net, img_list, group_size, img_size):
         #pred_mask=[F.interpolate(pred_mask[i].reshape(1,1,pred_mask[i].shape[-2],pred_mask[i].shape[-1]),size=(size,size),mode='bilinear').squeeze().numpy().astype(np.uint8) for i in range(5)]
         img_resize=[((group_img[i]-group_img[i].min())/(group_img[i].max()-group_img[i].min())*255).permute(1,2,0).contiguous().numpy().astype(np.uint8) 
                     for i in range(5)]
-        pred_mask=[crf_refine(img_resize[i],pred_mask[i].numpy().astype(np.uint8)) for i in range(5)]
+        pred_mask=[(img_resize[i],pred_mask[i].numpy().astype(np.uint8)) for i in range(5)]
         #for i in range(5):
         #    print(img_list[i].shape,pred_mask[i].shape)
         #pred_mask=[crf_refine(img_list[i],pred_mask[i]) for i in range(5)]