Add :json_polyfill when required

Author: kipcole9

lib/classification.ex

@@ -156,16 +156,16 @@ if ImageVision.bumblebee_configured?() do
     """
     @spec classifier(configuration :: Keyword.t()) ::
             {Nx.Serving, Keyword.t()} | {:error, Image.error()}
-    def classifier(classifier \\ Application.get_env(:image_vision, :classifier, [])) do
-      classifier = Keyword.merge(@default_classifier, classifier)
+    def classifier(configuration \\ Application.get_env(:image_vision, :classifier, [])) do
+      configuration = Keyword.merge(@default_classifier, configuration)
 
-      model = Keyword.fetch!(classifier, :model)
-      model_options = Keyword.fetch!(classifier, :model_options)
+      model = Keyword.fetch!(configuration, :model)
+      model_options = Keyword.fetch!(configuration, :model_options)
 
-      featurizer = Keyword.fetch!(classifier, :featurizer)
-      featurizer_options = Keyword.fetch!(classifier, :featurizer_options)
+      featurizer = Keyword.fetch!(configuration, :featurizer)
+      featurizer_options = Keyword.fetch!(configuration, :featurizer_options)
 
-      batch_size = Keyword.fetch!(classifier, :batch_size)
+      batch_size = Keyword.fetch!(configuration, :batch_size)
 
       case Image.Classification.serving(
              model,
@@ -178,7 +178,7 @@ if ImageVision.bumblebee_configured?() do
           {:error, error}
 
         serving ->
-          {Nx.Serving, serving: serving, name: classifier[:name], batch_timeout: 100}
+          {Nx.Serving, serving: serving, name: configuration[:name], batch_timeout: 100}
       end
     end
 
@@ -222,16 +222,16 @@ if ImageVision.bumblebee_configured?() do
     """
     @spec embedder(configuration :: Keyword.t()) ::
             {Nx.Serving, Keyword.t()} | {:error, Image.error()}
-    def embedder(embedder \\ Application.get_env(:image_vision, :embedder, [])) do
-      embedder = Keyword.merge(@default_embedder, embedder)
+    def embedder(configuration \\ Application.get_env(:image_vision, :embedder, [])) do
+      configuration = Keyword.merge(@default_embedder, configuration)
 
-      model = Keyword.fetch!(embedder, :model)
-      model_options = Keyword.fetch!(embedder, :model_options)
+      model = Keyword.fetch!(configuration, :model)
+      model_options = Keyword.fetch!(configuration, :model_options)
 
-      featurizer = Keyword.fetch!(embedder, :featurizer)
-      featurizer_options = Keyword.fetch!(embedder, :featurizer_options)
+      featurizer = Keyword.fetch!(configuration, :featurizer)
+      featurizer_options = Keyword.fetch!(configuration, :featurizer_options)
 
-      batch_size = Keyword.fetch!(embedder, :batch_size)
+      batch_size = Keyword.fetch!(configuration, :batch_size)
 
       case embedding_serving(
              model,
@@ -244,7 +244,7 @@ if ImageVision.bumblebee_configured?() do
           {:error, error}
 
         serving ->
-          {Nx.Serving, serving: serving, name: embedder[:name], batch_timeout: 100}
+          {Nx.Serving, serving: serving, name: configuration[:name], batch_timeout: 100}
       end
     end
 

lib/detection.ex

@@ -288,10 +288,10 @@ if ImageVision.ortex_configured?() do
     # by `:min_score`, scale boxes back to the original image, and
     # return them sorted by descending score.
 
-    defp postprocess(logits, pred_boxes, scale_x, scale_y, opts) do
-      original_width = Keyword.fetch!(opts, :original_width)
-      original_height = Keyword.fetch!(opts, :original_height)
-      min_score = Keyword.fetch!(opts, :min_score)
+    defp postprocess(logits, pred_boxes, scale_x, scale_y, options) do
+      original_width = Keyword.fetch!(options, :original_width)
+      original_height = Keyword.fetch!(options, :original_height)
+      min_score = Keyword.fetch!(options, :min_score)
 
       scores =
         logits[0]
@@ -338,11 +338,11 @@ if ImageVision.ortex_configured?() do
     # pixel coordinates on the original image. Clips to image bounds
     # and returns nil if the result has zero area.
 
-    defp cxcywh_to_xywh(cx, cy, w, h, opts) do
-      scale_x = Keyword.fetch!(opts, :scale_x)
-      scale_y = Keyword.fetch!(opts, :scale_y)
-      max_width = Keyword.fetch!(opts, :max_width)
-      max_height = Keyword.fetch!(opts, :max_height)
+    defp cxcywh_to_xywh(cx, cy, w, h, options) do
+      scale_x = Keyword.fetch!(options, :scale_x)
+      scale_y = Keyword.fetch!(options, :scale_y)
+      max_width = Keyword.fetch!(options, :max_width)
+      max_height = Keyword.fetch!(options, :max_height)
 
       input_size = @input_size
 

lib/segmentation.ex

@@ -560,10 +560,10 @@ if ImageVision.ortex_configured?() do
     end
 
     # Converts raw DETR-panoptic outputs into a list of segments.
-    defp detr_postprocess(logits, pred_masks, id2label, opts) do
-      min_score = Keyword.fetch!(opts, :min_score)
-      orig_w = Keyword.fetch!(opts, :orig_w)
-      orig_h = Keyword.fetch!(opts, :orig_h)
+    defp detr_postprocess(logits, pred_masks, id2label, options) do
+      min_score = Keyword.fetch!(options, :min_score)
+      orig_w = Keyword.fetch!(options, :orig_w)
+      orig_h = Keyword.fetch!(options, :orig_h)
 
       # logits: [1, 100, 251] → numerically-stable softmax, pick top class per query
       query_logits = logits[0]

mix.exs

@@ -71,7 +71,15 @@ defmodule ImageVision.MixProject do
       # --- Tooling ---
       {:ex_doc, "~> 0.18", only: [:release, :dev, :docs]},
       {:dialyxir, "~> 1.4", only: [:dev, :test], runtime: false}
-    ]
+    ] ++ maybe_json_polyfill()
+  end
+
+  defp maybe_json_polyfill do
+    if Code.ensure_loaded?(:json) do
+      []
+    else
+      [{:json_polyfill, "~> 0.2 or ~> 1.0"}]
+    end
   end
 
   defp package do