Upload package in yml file

jiaowoguanren0615 · Jun 15, 2024 · ecaf9b0 · ecaf9b0
1 parent 07e7f1f
commit ecaf9b0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/environment.yml b/environment.yml
@@ -3,6 +3,9 @@ dependencies:
   - python=3.9
   - pip
   - pip:
+      - onnx==1.13
+      - onnxoptimizer==0.3.13
+      - onnxruntime==1.18
       - matplotlib==3.5.1
       - numpy==1.23.0
       - opencv-contrib-python==4.7.0.72

diff --git a/onnx_validate.py b/onnx_validate.py
@@ -17,6 +17,7 @@
 import util.utils as utils
 
 
+
 parser = argparse.ArgumentParser(description='Pytorch ONNX Validation')
 parser.add_argument("--Kvasir_path", type=str, default='/mnt/d/MedicalSeg/Kvasir-SEG/',
                         help="path to Kvasir Dataset")