/
ViewController.swift
65 lines (42 loc) · 1.82 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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
//
// ViewController.swift
// OSXDeepLearningKitApp
//
// Created by Amund Tveit on 15/02/16.
// Copyright © 2016 DeepLearningKit. All rights reserved.
//
import Cocoa
class ViewController: NSViewController {
var deepNetwork: DeepNetwork!
override func viewDidLoad() {
super.viewDidLoad()
}
override func viewDidAppear() {
deepNetwork = DeepNetwork()
// conv1.json contains a cifar 10 image of a cat
let conv1Layer = deepNetwork.loadJSONFile("conv1")!
let image: [Float] = conv1Layer["input"] as! [Float]
var randomimage = createFloatNumbersArray(image.count)
for i in 0..<randomimage.count {
randomimage[i] = Float(arc4random_uniform(1000))
}
let imageShape:[Float] = [1.0, 3.0, 32.0, 32.0]
let caching_mode = false
// 0. load network in network model
deepNetwork.loadDeepNetworkFromJSON("nin_cifar10_full", inputImage: image, inputShape: imageShape, caching_mode:caching_mode)
// 1. classify image (of cat)
deepNetwork.classify(image)
// 2. reset deep network and classify random image
deepNetwork.loadDeepNetworkFromJSON("nin_cifar10_full", inputImage: randomimage, inputShape: imageShape,caching_mode:caching_mode)
deepNetwork.classify(randomimage)
// 3. reset deep network and classify cat image again
deepNetwork.loadDeepNetworkFromJSON("nin_cifar10_full", inputImage: image, inputShape: imageShape,caching_mode:caching_mode)
deepNetwork.classify(image)
exit(0)
}
override var representedObject: AnyObject? {
didSet {
// Update the view, if already loaded.
}
}
}