Update inference_img.py

inference_img.py

@@ -5,7 +5,7 @@ import argparse
 from torch.nn import functional as F
 import warnings
 
-def main():
+
         
     warnings.filterwarnings("ignore")
     
@@ -103,5 +103,4 @@ def main():
         else:
             cv2.imwrite('/output/img{}.png'.format(i), (img_list[i][0] * 255).byte().cpu().numpy().transpose(1, 2, 0)[:h, :w])
 
-if __name__ == "__main__":
-    main()
+