update read json file code

Author: WZMIAOMIAO

pytorch_classification/ConvNeXt/predict.py

@@ -35,8 +35,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = create_model(num_classes=num_classes).to(device)

pytorch_classification/Test10_regnet/predict.py

@@ -32,8 +32,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = create_regnet(model_name="RegNetY_400MF", num_classes=5).to(device)

pytorch_classification/Test11_efficientnetV2/predict.py

@@ -37,8 +37,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = create_model(num_classes=5).to(device)

pytorch_classification/Test2_alexnet/predict.py

@@ -32,8 +32,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = AlexNet(num_classes=5).to(device)

pytorch_classification/Test3_vggnet/predict.py

@@ -31,8 +31,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = vgg(model_name="vgg16", num_classes=5).to(device)

pytorch_classification/Test4_googlenet/predict.py

@@ -31,8 +31,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = GoogLeNet(num_classes=5, aux_logits=False).to(device)

pytorch_classification/Test5_resnet/predict.py

@@ -32,8 +32,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = resnet34(num_classes=5).to(device)

pytorch_classification/Test6_mobilenet/predict.py

@@ -32,8 +32,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = MobileNetV2(num_classes=5).to(device)

pytorch_classification/Test7_shufflenet/predict.py

@@ -32,8 +32,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = shufflenet_v2_x1_0(num_classes=5).to(device)

pytorch_classification/Test8_densenet/predict.py

@@ -32,8 +32,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = densenet121(num_classes=5).to(device)

pytorch_classification/Test9_efficientNet/predict.py

@@ -42,8 +42,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = create_model(num_classes=5).to(device)

pytorch_classification/swin_transformer/predict.py

@@ -33,8 +33,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = create_model(num_classes=5).to(device)

pytorch_classification/vision_transformer/predict.py

@@ -32,8 +32,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = create_model(num_classes=5, has_logits=False).to(device)

tensorflow_classification/ConvNeXt/predict.py

@@ -35,8 +35,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = create_model(num_classes=num_classes)

tensorflow_classification/Test11_efficientnetV2/predict.py

@@ -40,8 +40,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = create_model(num_classes=num_classes)

tensorflow_classification/Test2_alexnet/predict.py

@@ -31,8 +31,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = AlexNet_v1(num_classes=5)

tensorflow_classification/Test3_vgg/predict.py

@@ -31,8 +31,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = vgg("vgg16", im_height=im_height, im_width=im_width, num_classes=num_classes)

tensorflow_classification/Test4_goolenet/predict.py

@@ -31,8 +31,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     model = GoogLeNet(class_num=5, aux_logits=False)
     model.summary()

tensorflow_classification/Test5_resnet/predict.py

@@ -37,8 +37,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     feature = resnet50(num_classes=num_classes, include_top=False)

tensorflow_classification/Test6_mobilenet/predict.py

@@ -34,8 +34,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     feature = MobileNetV2(include_top=False)

tensorflow_classification/Test7_shuffleNet/predict.py

@@ -36,8 +36,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = shufflenet_v2_x1_0(num_classes=num_classes)

tensorflow_classification/Test9_efficientNet/predict.py

@@ -41,8 +41,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = create_model(num_classes=num_classes)

tensorflow_classification/swin_transformer/predict.py

@@ -35,8 +35,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = create_model(num_classes=num_classes)

tensorflow_classification/vision_transformer/predict.py

@@ -35,8 +35,8 @@ def main():
     json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
 
-    json_file = open(json_path, "r")
-    class_indict = json.load(json_file)
+    with open(json_path, "r") as f:
+        class_indict = json.load(f)
 
     # create model
     model = create_model(num_classes=num_classes, has_logits=False)