Update app.py

app.py

@@ -84,9 +84,9 @@ def test(gpu_id, net, img_list, group_size, img_size):
         group_img=torch.rand(5,3,224,224)
         for i in range(5):
           group_img[i]=img_transform(Image.fromarray(img_list[i]))
-        _,pred_mask=net(group_img)
+        _,pred_mask=net(group_img*1)
         pred_mask=(pred_mask.detach().squeeze()*255).numpy().astype(np.uint8)
-        pred_mask=[crf_refine(img_list[i],pred_mask[i]) for i in range(5)]
+        pred_mask=[crf_refine(((group_img[i]-group_img[i].min())/(group_img[i].max()-group_img[i].min())*255).numpy().astype(np.uint8),pred_mask[i]) for i in range(5)]
         print(pred_mask[0].shape)
         result = [Image.fromarray((torch.from_numpy(pred_mask[i]).unsqueeze(2).repeat(1,1,3)).numpy()) for i in range(5)]
         #w, h = 224,224#Image.open(image_list[i][j]).size