-
Notifications
You must be signed in to change notification settings - Fork 4
/
checkpoint_test.go
82 lines (70 loc) · 1.77 KB
/
checkpoint_test.go
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
package tensorflow
// import (
// "io/ioutil"
// "os"
// "path/filepath"
// "testing"
// "github.com/k0kubun/pp"
// "github.com/rai-project/config"
// "github.com/rai-project/downloadmanager"
// "github.com/stretchr/testify/assert"
// tf "github.com/tensorflow/tensorflow/tensorflow/go"
// )
// func TestCheckpointReader(t *testing.T) {
// url := "http://download.tensorflow.org/models/image/imagenet/inception-v3-2016-03-01.tar.gz"
// targetDir := filepath.Join(config.App.TempDir, "testing", "inception")
// path, err := downloadmanager.Download(url, targetDir)
// assert.NoError(t, err)
// if err != nil {
// return
// }
// assert.NotEmpty(t, path)
// path = filepath.Dir(path)
// pp.Println(path)
// metaPath := filepath.Join(path, "model.ckpt-157585")
// f, err := os.Open(metaPath)
// assert.NoError(t, err)
// if err != nil {
// return
// }
// defer f.Close()
// rd, err := ioutil.ReadAll(f)
// if err != nil {
// return
// }
// if true {
// model, err := tf.LoadSavedModel("/tmp/carml/testing/mobilenet/mobilenet_v1_0.25_128.ckpt", []string{"checkpoint"}, nil)
// assert.NoError(t, err)
// _ = model
// return
// }
// if false {
// graph := tf.NewGraph()
// err = graph.Import(rd, "")
// assert.NoError(t, err)
// if err != nil {
// return
// }
// return
// }
// // if false {
// // s := op.NewScope()
// // tf.RestoreV2(s, op.Const(s, targetDir))
// // }
// info := &MetaGraphDef{}
// err = info.Unmarshal(rd)
// assert.NoError(t, err)
// if err != nil {
// return
// }
// pp.Println(info.GetMetaInfoDef())
// pp.Println(info.GetAssetFileDef())
// }
// func TestMain(m *testing.M) {
// config.Init(
// config.AppName("carml"),
// config.DebugMode(true),
// config.VerboseMode(true),
// )
// os.Exit(m.Run())
// }