try with a new path value (#69)

* try with new path value * aclconfigpath failed, only try with new model path * aclconfigpath failed, only try with new model path * try with new model path * improve the format
amosproj · May 26, 2021 · 37b7882 · 37b7882
1 parent e017588
commit 37b7882
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/colorization/src/colorize_process.py b/colorization/src/colorize_process.py
@@ -85,6 +85,9 @@ def InitResource(self):
             on success this function returns 0
             on failure this function returns 1
         """
+        #  ACLCONFIGPATH = os.path.join(os.path.abspath(os.path.dirname(__file__)),
+        #                             '../acl.json')
+        #  ret = acl.init(ACLCONFIGPATH)
         ret = acl.init()  # no configuration.info as argument
         if ret != acl_constants.ACL_ERROR_NONE:
             print("Acl init failed")

diff --git a/colorization/src/pipeline.py b/colorization/src/pipeline.py
@@ -1,6 +1,7 @@
 from colorize_process import ColorizeProcess
 # from utils import CopyDataDeviceToHost
 import numpy
+import os
 # import cv2
 # from Data.data import *
 # import splitVideo
@@ -51,7 +52,8 @@ def colorize_image(image_path_input, image_path_output):
 
     kModelWidth = numpy.uint32(224)
     kModelHeight = numpy.uint32(224)
-    KMODELPATH = "../model/colorization.om"
+    KMODELPATH = os.path.join(os.path.abspath(os.path.dirname(__file__)),
+                              "../../model/colorization.om")
     colorize = ColorizeProcess(KMODELPATH, kModelWidth, kModelHeight)
     ret = colorize.Init()
     if ret == FAILED: