diff --git a/project/Dependencies.scala b/project/Dependencies.scala index e78d580..42e85d0 100644 --- 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/recognizer/src/main/scala/com/github/baklanovsoft/imagehosting/recognizer/ObjectDetection.scala b/recognizer/src/main/scala/com/github/baklanovsoft/imagehosting/recognizer/ObjectDetection.scala index d35d8ce..f7817d0 100644 --- a/recognizer/src/main/scala/com/github/baklanovsoft/imagehosting/recognizer/ObjectDetection.scala +++ b/recognizer/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"