From 9a77b9807fe681da9588bfcb97321f2a6f9a2d73 Mon Sep 17 00:00:00 2001
From: Nattapon Jaroenchai <b_nj18@live.com>
Date: Thu, 2 Nov 2023 17:45:21 -0500
Subject: [PATCH] remove jsonPath
---
inference.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/inference.py b/inference.py
index 4a619f2..f5015bf 100644
--- a/inference.py
+++ b/inference.py
@@ -210,7 +210,7 @@ def save_results(prediction, map_name, legend, outputPath):
### Waiting for georeferencing data
# This section will be used in future releases to save georeferenced images.
- ### Waiting for georeferencing data
+ ## Waiting for georeferencing data
# with rasterio.open(map_img_path) as src:
# metadata = src.meta
@@ -334,7 +334,6 @@ if __name__ == "__main__":
# Command-line interface setup
parser = argparse.ArgumentParser(description="Perform inference on a given map.")
parser.add_argument("--mapPath", required=True, help="Path to the hdf5 file.")
- parser.add_argument("--jsonPath", required=True, help="Path to the JSON file that contain positions of legends.")
parser.add_argument("--featureType", choices=["Polygon", "Point", "Line", "All"], default="Polygon", help="Type of feature to detect. Three options, Polygon, Point, and, Line")
parser.add_argument("--outputPath", required=True, help="Path to save the inference results. ")
parser.add_argument("--modelPath", default="./inference_model/Unet-attentionUnet.h5", help="Path to the trained model. Default is './inference_model/Unet-attentionUnet.h5'.")
@@ -343,4 +342,4 @@ if __name__ == "__main__":
main(args)
# Test command
-# python inference.py --mapPath "/projects/bbym/shared/data/commonPatchData/256/OK_250K.hdf5" --jsonPath "" --featureType "Polygon" --outputPath "/projects/bbym/nathanj/attentionUnet/infer_results" --modelPath "/projects/bbym/nathanj/attentionUnet/inference_model/Unet-attentionUnet.h5"
\ No newline at end of file
+# python inference.py --mapPath "/projects/bbym/shared/data/commonPatchData/256/OK_250K.hdf5" --featureType "Polygon" --outputPath "/projects/bbym/nathanj/attentionUnet/infer_results" --modelPath "/projects/bbym/nathanj/attentionUnet/inference_model/Unet-attentionUnet.h5"
\ No newline at end of file
--
GitLab