conversion function for Makie

IanButterworth · Sep 19, 2019 · 4798772 · 4798772
1 parent 7791c9c
commit 4798772
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/makierendering.jl b/src/makierendering.jl
@@ -29,3 +29,6 @@ function renderResult(img::Array{Float32}, predictions::Vector{YOLO.PredictLabel
     save_file != "" && Makie.save(save_file, scene)
     return scene
 end
+
+renderResult(img::KnetArray{Float32}, predictions::Vector{YOLO.PredictLabel}, settings::YOLO.Settings; save_file::String = "") =
+renderResult(Array{Float32}(img), predictions = predictions, settings = settings, save_file = save_file)