diff --git a/lib/ex_vision/classification/mobilenet_v3_small.ex b/lib/ex_vision/classification/mobilenet_v3_small.ex index 5f4358c..ac30093 100644 --- a/lib/ex_vision/classification/mobilenet_v3_small.ex +++ b/lib/ex_vision/classification/mobilenet_v3_small.ex @@ -6,7 +6,7 @@ defmodule ExVision.Classification.MobileNetV3Small do """ use ExVision.Model.Definition.Ortex, model: "mobilenetv3small-classifier.onnx", - categories: "assets/categories/imagenet_v2_categories.json" + categories: "priv/categories/imagenet_v2_categories.json" require Bunch.Typespec alias ExVision.Utils diff --git a/lib/ex_vision/detection/fasterrcnn_resnet50_fpn.ex b/lib/ex_vision/detection/fasterrcnn_resnet50_fpn.ex index 0476b15..219572a 100644 --- a/lib/ex_vision/detection/fasterrcnn_resnet50_fpn.ex +++ b/lib/ex_vision/detection/fasterrcnn_resnet50_fpn.ex @@ -4,7 +4,7 @@ defmodule ExVision.Detection.FasterRCNN_ResNet50_FPN do """ use ExVision.Model.Definition.Ortex, model: "fasterrcnn_resnet50_fpn_detector.onnx", - categories: "assets/categories/coco_categories.json" + categories: "priv/categories/coco_categories.json" use ExVision.Detection.GenericDetector diff --git a/lib/ex_vision/detection/ssdlite320_mobilenetv3.ex b/lib/ex_vision/detection/ssdlite320_mobilenetv3.ex index 043ddb1..dc48372 100644 --- a/lib/ex_vision/detection/ssdlite320_mobilenetv3.ex +++ b/lib/ex_vision/detection/ssdlite320_mobilenetv3.ex @@ -4,7 +4,7 @@ defmodule ExVision.Detection.Ssdlite320_MobileNetv3 do """ use ExVision.Model.Definition.Ortex, model: "ssdlite320_mobilenetv3_detector.onnx", - categories: "assets/categories/coco_categories.json" + categories: "priv/categories/coco_categories.json" use ExVision.Detection.GenericDetector diff --git a/lib/ex_vision/segmentation/deep_lab_v3_mobilenet_v3.ex b/lib/ex_vision/segmentation/deep_lab_v3_mobilenet_v3.ex index c2a6735..2ef85d7 100644 --- a/lib/ex_vision/segmentation/deep_lab_v3_mobilenet_v3.ex +++ b/lib/ex_vision/segmentation/deep_lab_v3_mobilenet_v3.ex @@ -4,7 +4,7 @@ defmodule ExVision.Segmentation.DeepLabV3_MobileNetV3 do """ use ExVision.Model.Definition.Ortex, model: "deeplab_v3_mobilenetv3_segmentation.onnx", - categories: "assets/categories/coco_with_voc_labels_categories.json" + categories: "priv/categories/coco_with_voc_labels_categories.json" @type output_t() :: %{category_t() => Nx.Tensor.t()} diff --git a/mix.exs b/mix.exs index d7be525..8809fdf 100644 --- a/mix.exs +++ b/mix.exs @@ -1,10 +1,8 @@ defmodule ExVision.Mixfile do use Mix.Project - # TODO: make sure that the links here are correct - - @version "0.1.0" - @github_url "https://github.com/software-mansion-labs/ex_vision" + @version "0.1.0-rc1" + @github_url "https://github.com/membraneframework-labs/ex_vision/" def project do [ diff --git a/assets/categories/coco_categories.json b/priv/categories/coco_categories.json similarity index 100% rename from assets/categories/coco_categories.json rename to priv/categories/coco_categories.json diff --git a/assets/categories/coco_with_voc_labels_categories.json b/priv/categories/coco_with_voc_labels_categories.json similarity index 100% rename from assets/categories/coco_with_voc_labels_categories.json rename to priv/categories/coco_with_voc_labels_categories.json diff --git a/assets/categories/imagenet_v2_categories.json b/priv/categories/imagenet_v2_categories.json similarity index 100% rename from assets/categories/imagenet_v2_categories.json rename to priv/categories/imagenet_v2_categories.json