-
Notifications
You must be signed in to change notification settings - Fork 0
/
ViewController.swift
49 lines (37 loc) · 1.35 KB
/
ViewController.swift
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
//
// ViewController.swift
// MachineLearningApp
//
// Created by cagdas on 18/04/2018.
// Copyright © 2018 cagdas. All rights reserved.
//
import UIKit
class ViewController: UIViewController {
@IBOutlet weak var categoryLabel: UILabel!
@IBOutlet var imageTapped: UISwipeGestureRecognizer!
let model = GoogLeNetPlaces()
override func viewDidLoad() {
super.viewDidLoad()
// Do any additional setup after loading the view, typically from a nib.
}
@IBAction func imageTapped(_ sender: UITapGestureRecognizer) {
let imageView = sender.view as? UIImageView
print("imageTapped")
if let imageToAnalyse = imageView?.image {
if let sceneLabelString = sceneLabel(forImage: imageToAnalyse) {
categoryLabel.text = sceneLabelString
}
}
}
func sceneLabel (forImage image:UIImage) -> String? {
if let pixelBuffer = ImageProcessor.pixelBuffer(forImage: image.cgImage!) {
guard let scene = try? model.prediction(sceneImage: pixelBuffer) else {fatalError("Unexpected runtime error")}
return scene.sceneLabel
}
return nil
}
override func didReceiveMemoryWarning() {
super.didReceiveMemoryWarning()
// Dispose of any resources that can be recreated.
}
}