add chart

jcao-ai · Dec 11, 2017 · 28881bf · 28881bf
1 parent 3fa38c8
commit 28881bf
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 1 deletion.
diff --git a/DemoImages/chart_of_beam_size b/DemoImages/chart_of_beam_size
diff --git a/DemoImages/showAndTell.png b/DemoImages/showAndTell.png
diff --git a/ShowAndTell/ViewController.swift b/ShowAndTell/ViewController.swift
@@ -69,7 +69,7 @@ class ViewController: UIViewController {
     @IBAction func predict() {
         textView.text = nil
         let startTime = Date()
-        let results = showAndTell.predict(image: self.currentImage, beamSize: 3, maxWordNumber: 30)
+        let results = showAndTell.predict(image: self.currentImage, beamSize: 7, maxWordNumber: 30)
         GSMessage.showMessageAddedTo("Time elapsed：\(Date().timeIntervalSince(startTime) * 1000)ms", type: .info, options: nil, inView: self.view, inViewController: self)
         textView.text = results.sorted(by: {$0.score > $1.score}).map({
             var x = $0.readAbleSentence.suffix($0.readAbleSentence.count - 1)