From 3fd7b3c086d22a85b9e2b383018310023c970060 Mon Sep 17 00:00:00 2001
From: Piero Toffanin <pt@masseranolabs.com>
Date: Tue, 6 Sep 2022 16:53:11 -0400
Subject: [PATCH] Remove dependency to onnx

---
 opendm/skyremoval/skyfilter.py | 12 ------------
 requirements.txt               |  3 +--
 2 files changed, 1 insertion(+), 14 deletions(-)

diff --git a/opendm/skyremoval/skyfilter.py b/opendm/skyremoval/skyfilter.py
index d82f4b7d..56fee610 100644
--- a/opendm/skyremoval/skyfilter.py
+++ b/opendm/skyremoval/skyfilter.py
@@ -3,7 +3,6 @@ import time
 import numpy as np
 import cv2
 import os
-import onnx
 import onnxruntime as ort
 from .guidedfilter import guided_filter
 from opendm import log
@@ -27,17 +26,6 @@ class SkyFilter():
     
     def load_model(self):
         log.ODM_INFO(' -> Loading the model')
-        onnx_model = onnx.load(self.model)
-
-        # Check the model
-        try:
-            onnx.checker.check_model(onnx_model)
-        except onnx.checker.ValidationError as e:
-            log.ODM_INFO(' !> The model is invalid: %s' % e)
-            raise
-        else:
-            log.ODM_INFO(' ?> The model is valid!')
-
         self.session = ort.InferenceSession(self.model, providers=[provider])     
 
 
diff --git a/requirements.txt b/requirements.txt
index b05b3f61..d74435dd 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -29,5 +29,4 @@ scipy==1.5.4
 xmltodict==0.12.0
 fpdf2==2.4.6
 Shapely==1.7.1
-onnx==1.12.0
-onnxruntime==1.11.1
\ No newline at end of file
+onnxruntime==1.12.1
\ No newline at end of file