Upload app.py

app.py

@@ -104,12 +104,11 @@ def single_scale_test(image):
     #torchvision.utils.save_image(1 - all_res, osp.join('RCFPyTorch0/results/RCF', 'result.jpg'))
     fuse_res = torch.squeeze(results[1].detach()).cpu().numpy()
     fuse_res = ((1 - fuse_res) * 255).astype('uint8')
-    cv2.imwrite(osp.join("RCFPyTorch0/results/RCF", 'result_ss.png'), fuse_res)
+    #cv2.imwrite(osp.join("RCFPyTorch0/results/RCF", 'result_ss.png'), fuse_res)
     #print('\rRunning single-scale test [%d/%d]' % (idx + 1, len(test_loader)), end='')
     #print('Running single-scale test done')
-    output = Image.open(os.path.join('RCFPyTorch0/results/RCF', 'result_ss.png'))
-    #output = Image.fromarray(((fuse_res).astype('uint8')), mode='RGB')
-    print(type(output))
+    #output = Image.open(os.path.join('RCFPyTorch0/results/RCF', 'result_ss.png'))
+    output = Image.fromarray((fuse_res).astype('uint8'))
     return output
 
 parser = argparse.ArgumentParser(description='PyTorch Testing')