experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

go: Make example easier to try out.

Browse Source 
godoc.org translates full-file examples into a "main" package. This change
makes the example so that the generated "main" code can be copy-pasted and thus
used to try out a standalone binary.

Another step towards #10
Change: 134709944
This commit is contained in:
Asim Shankar 2016-09-29 13:34:47 -08:00 committed by TensorFlower Gardener
parent e6be8a4b24
commit 2f6e25df47
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tensorflow/go/example_inception_inference_test.go
View File

@ -12,7 +12,7 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
package tensorflow package tensorflow_test
import ( import (
"bufio" "bufio"
@ -22,6 +22,8 @@ import (
"io/ioutil" "io/ioutil"
"log" "log"
"os" "os"
tf "github.com/tensorflow/tensorflow/tensorflow/go"
) )
func Example() { func Example() {
@ -69,13 +71,13 @@ func Example() {
} }
// Construct an in-memory graph from the serialized form. // Construct an in-memory graph from the serialized form.
graph := NewGraph() graph := tf.NewGraph()
if err := graph.Import(model, ""); err != nil { if err := graph.Import(model, ""); err != nil {
log.Fatal(err) log.Fatal(err)
} }
// Create a session for inference over graph. // Create a session for inference over graph.
session, err := NewSession(graph, nil) session, err := tf.NewSession(graph, nil)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
@ -90,10 +92,10 @@ func Example() {
log.Fatal(err) log.Fatal(err)
} }
output, err := session.Run( output, err := session.Run(
map[Output]*Tensor{ map[tf.Output]*tf.Tensor{
graph.Operation("input").Output(0): tensor, graph.Operation("input").Output(0): tensor,
}, },
[]Output{ []tf.Output{
graph.Operation("output").Output(0), graph.Operation("output").Output(0),
}, },
nil) nil)
@ -134,7 +136,7 @@ func printBestLabel(probabilities []float32, labelsFile string) {
// Given an image stored in filename, returns a Tensor which is suitable for // Given an image stored in filename, returns a Tensor which is suitable for
// providing the image data to the pre-defined model. // providing the image data to the pre-defined model.
func makeTensorFromImageForInception(filename string) (*Tensor, error) { func makeTensorFromImageForInception(filename string) (*tf.Tensor, error) {
const ( const (
// Some constants specific to the pre-trained model at: // Some constants specific to the pre-trained model at:
// https://storage.googleapis.com/download.tensorflow.org/models/inception5h.zip // https://storage.googleapis.com/download.tensorflow.org/models/inception5h.zip
@ -179,5 +181,5 @@ func makeTensorFromImageForInception(filename string) (*Tensor, error) {
ret[0][y][x][2] = float32((int(r>>8) - Mean)) / Std ret[0][y][x][2] = float32((int(r>>8) - Mean)) / Std
} }
} }
return NewTensor(ret) return tf.NewTensor(ret)
} }