Update main_test_SRMNet.py

main_test_SRMNet.py

@@ -30,7 +30,7 @@ def main():
     args = parser.parse_args()
 
     # Prepare testing data
-    files = natsorted(glob(os.path.join(args.inp_dir, '*.PNG'))
+    files = natsorted(glob(os.path.join(inp_dir, '*.JPG')) + glob(os.path.join(inp_dir, '*.PNG')))
     if len(files) == 0:
         raise Exception("\nNo images in {} \nPlease enter the following tasks: \n\n{}".format(inp_dir, '\n'.join(tasks)))
     os.makedirs(args.result_dir, exist_ok=True)