Linux/MacOC Path ~

realglobe-Inc · Aug 7, 2024 · 1ad6431 · 1ad6431
1 parent a152525
commit 1ad6431
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/predict.py b/predict.py
@@ -134,7 +134,7 @@ def sorted_glob(pattern):
         param = json.load(pf)
 
     main(
-        img_pattern=os.path.join(param['FilePath']['InputImagePath'], '*'),
-        weights_path=os.path.join(param['FilePath']['CheckpointsPath'], 'fpn_inception.h5'),
-        out_dir=param['FilePath']['OutputImagePath']
+        img_pattern=os.path.join(os.path.expanduser(param['FilePath']['InputImagePath']), '*'),
+        weights_path=os.path.join(os.path.expanduser(param['FilePath']['CheckpointsPath']), 'fpn_inception.h5'),
+        out_dir=os.path.expanduser(param['FilePath']['OutputImagePath'])
     )