predict.clj

(defn predict [img-url show?]
  (let [mod (m/load-checkpoint {:prefix (str model-dir "/resnet-152") :epoch 0})
        labels (-> (slurp (str model-dir "/synset.txt"))
                   (string/split #"\n"))
        nd-img (get-image img-url show?)
        prob (-> mod
                 (m/bind {:for-training false 
                          :data-shapes [{:name "data" :shape [1 num-channels h w]}]})
                 (m/forward {:data [nd-img]})
                 (m/outputs)
                 (ffirst))
        prob-with-labels (mapv (fn [p l] {:prob p :label l})
                               (ndarray/->vec prob)
                               labels)]
    (->> (sort-by :prob prob-with-labels)
         (reverse)
         (take 5))))