Merge pull request drivendataorg#14 from AliGebily/segmentation-model…

…s-package add segmentation-models==1.0.1
fischcheng · Sep 14, 2021 · e5be143 · e5be143
2 parents 6760233 + ebd9565
commit e5be143
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/runtime/environment-cpu.yml b/runtime/environment-cpu.yml
@@ -43,3 +43,4 @@ dependencies:
     - mmcv==1.3.9
     - mmsegmentation==0.16.0
     - tensorflow-cpu==2.6.0
+    - segmentation-models==1.0.1
diff --git a/runtime/environment-gpu.yml b/runtime/environment-gpu.yml
@@ -46,3 +46,4 @@ dependencies:
     - mmcv-full==1.3.9
     - mmsegmentation==0.16.0
     - tensorflow-gpu==2.6.0
+    - segmentation-models==1.0.1