init

p2c/test_onnx.py

@@ -9,6 +9,7 @@ class Photo2Cartoon:
     def __init__(self):
         self.pre = Preprocess()
         curPath = os.path.abspath(os.path.dirname(__file__))
+        print(os.path.join(curPath, 'models/photo2cartoon_weights.onnx'))
         # assert os.path.exists('./models/photo2cartoon_weights.onnx'), "[Step1: load weights] Can not find 'photo2cartoon_weights.onnx' in folder 'models!!!'"
         self.session = onnxruntime.InferenceSession(os.path.join(curPath, 'models/photo2cartoon_weights.onnx'))
         print('[Step1: load weights] success!')