Make added functionalities public

software-mansion-labs · Jul 4, 2024 · 294a8cd · 294a8cd
1 parent c973b8b
commit 294a8cd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/ex_vision/utils.ex b/lib/ex_vision/utils.ex
@@ -156,7 +156,7 @@ defmodule ExVision.Utils do
     process_name |> batched_run([input]) |> hd()
   end
 
-  defp process_bbox(bbox, scales, axes \\ [0]) do
+  def process_bbox(bbox, scales, axes \\ [0]) do
     bbox
     |> Nx.squeeze(axes: axes)
     |> Nx.multiply(scales)
@@ -165,7 +165,7 @@ defmodule ExVision.Utils do
     |> Nx.to_list()
   end
 
-  defp unbatch(batched_value, axes \\ [0]) do
+  def unbatch(batched_value, axes \\ [0]) do
     batched_value |> Nx.squeeze(axes: axes) |> Nx.to_list()
   end
 end