tensorflow-go-examples
tensorflow-go-examples copied to clipboard
Examples of using TensorFlow Go API for different Deep Learning models
TensorFlow Go Examples
This repo contains end-to-end examples of model inference using TensorFlow Go API. More are coming. Contributions with similar simple examples (or by further improving existing examples) are highly welcome. Please file a pull request or an issue on GitHub.
Models
-
Image Classification: Classify the main object category within an image.
-
Image Object Detection: Identify the object category and locate the position using a bounding box for every known object within an image.
-
Image Instance Segmentation: Identify each object instance of each pixel for every known object within an image. Labels are instance-aware.
-
image Semantic Segmentation: Identify the object category of each pixel for every known object within an image. Labels are class-aware.
-
image Enhancement
TensorFlow Go API
Refer to Install TensorFlow for Go.
To inspect pre-trained frozen graphs for input and output tensor names
References
- TensorFlow Go package
- TensorFlow model repo