IHP-32: remove mxnet since it will be deprecated

Baklanov-Soft · Sep 28, 2024 · 60996b2 · 60996b2
1 parent b43980b
commit 60996b2
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 17 deletions.
diff --git a/project/Dependencies.scala b/project/Dependencies.scala
@@ -47,9 +47,6 @@ object Dependencies {
 
   val djl = Seq(
     "ai.djl"         % "api"               % Versions.djl,
-    // mxnet is used in object detection for embedded vgg16
-    "ai.djl.mxnet"   % "mxnet-model-zoo"   % Versions.djl,
-    "ai.djl.mxnet"   % "mxnet-engine"      % Versions.djl,
     // pytorch for nsfw detection
     "ai.djl.pytorch" % "pytorch-engine"    % Versions.djl,
     "ai.djl.pytorch" % "pytorch-model-zoo" % Versions.djl

diff --git a/...izer/src/main/scala/com/github/baklanovsoft/imagehosting/recognizer/ObjectDetection.scala b/...izer/src/main/scala/com/github/baklanovsoft/imagehosting/recognizer/ObjectDetection.scala
@@ -24,21 +24,8 @@ trait ObjectDetection[F[_]] {
 object ObjectDetection {
 
   private object Engines {
-
-    // mx net has a lot of included models so it's better to use in detection
-    object MxNet {
-      val name = "MXNet"
-
-      object Models {
-        val darknet53 = "darknet53"
-        val mobileNet = "mobilenet1.0"
-        val resnet50  = "resnet50"
-        val vgg16     = "vgg16"
-      }
-    }
-
     object PyTorch {
-      val name = "MXNet"
+      val name = "PyTorch"
 
       object Models {
         val resnet50 = "resnet50"