back to ort, more options used

app.py

@@ -45,10 +45,12 @@ def load_model(model: str, activation: bool=True):
     if activation:
         model += '_w_activation'
     
-    sess_options = ort.SessionOptions()
-    sess_options.intra_op_num_threads = 1
-    # ort_session = ort.InferenceSession(model_path + '%s.onnx' % (model), sess_options)
-    ort_session = ORTModel(model_path + '%s.onnx' % (model))
+    options = ort.SessionOptions()
+    options.intra_op_num_threads = 1
+    options.graph_optimization_level = ort.GraphOptimizationLevel.ORT_ENABLE_ALL
+    provider = "CPUExecutionProvider"
+    ort_session = ort.InferenceSession(model_path + '%s.onnx' % (model), options, providers=[provider])
+    # ort_session = ORTModel.load_model(model_path + '%s.onnx' % (model))
     
     return ort_session
 
@@ -58,9 +60,9 @@ def get_activations(intermediate_model, image: list,
     '''Gets activations for a given input image'''
     
     
-    # input_name = intermediate_model.get_inputs()[0].name
-    # outputs = intermediate_model.run(None, {input_name: image})
-    outputs = intermediate_model(image)
+    input_name = intermediate_model.get_inputs()[0].name
+    outputs = intermediate_model.run(None, {input_name: image})
+    # outputs = intermediate_model(image)
     
     output_1 = outputs[1]
     output_2 = outputs[2]