gigasquid commented on a change in pull request #14533: [clojure][image] add
draw-bounding-box interop
URL: https://github.com/apache/incubator-mxnet/pull/14533#discussion_r270466436
##########
File path: contrib/clojure-package/src/org/apache/clojure_mxnet/image.clj
##########
@@ -199,3 +200,68 @@
[input]
(util/validate! ::to-image-ndarray input "Invalid input array")
(Image/toImage input))
+
+(s/def ::buffered-image #(instance? BufferedImage %))
+(s/def ::xmin integer?)
+(s/def ::xmax integer?)
+(s/def ::ymin integer?)
+(s/def ::ymax integer?)
+(s/def ::coordinate (s/keys :req-un [::xmin ::xmax ::ymin ::ymax]))
+(s/def ::coordinates (s/coll-of ::coordinate))
+(s/def ::names (s/nilable (s/coll-of string?)))
+(s/def ::stroke (s/and integer? pos?))
+(s/def ::font-size-mult (s/and float? pos?))
+(s/def ::transparency (s/and float? #(<= 0.0 % 1.0)))
+(s/def ::coordinates-names
+ (fn [[coordinates names]] (= (count coordinates) (count names))))
+
+(defn- convert-coordinate
+ "Convert bounding box coordinate to Scala correct types."
+ [{:keys [xmin xmax ymin ymax]}]
+ {:xmin (int xmin)
+ :xmax (int xmax)
+ :ymin (int ymin)
+ :ymax (int ymax)})
+
+(defn draw-bounding-box!
+ "Draw bounding boxes on `buffered-image` and Mutate the input image.
+ `buffered-image`: BufferedImage
+ `coordinates`: collection of {:xmin int :xmax int :ymin int :ymax int}
+ `font-size-mult`: positive float - Font size multiplier
+ `names`: collection of strings - List of names for the bounding boxes
+ `stroke`: positive integer - thickness of the bounding box
+ `transparency`: float in (0.0, 1.0) - Transparency of the bounding box
+ returns: Modified `buffered-image`
+ Ex:
Review comment:
I really like the examples in the docstring
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
