Skip to content
This repository was archived by the owner on Sep 17, 2019. It is now read-only.

Commit c770d67

Browse files
committed
Changed to appropriate names
1 parent 3fd790a commit c770d67

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/main/scala/tensorflow/TensorFlowExample.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ package tensorflow
22

33
import java.nio.file.{Files, Paths}
44

5-
import tensorflow.model.ImageNet
5+
import tensorflow.model.InceptionV3
66

77
object TensorFlowExample {
88

@@ -12,7 +12,7 @@ object TensorFlowExample {
1212
val jpgAsBytes = Files.readAllBytes(Paths.get(jpgFile))
1313

1414
// define the model
15-
val model = new ImageNet("model")
15+
val model = new InceptionV3("model")
1616

1717
// initialize TensorFlowProvider
1818
val provider = new TensorFlowProvider(model)

src/main/scala/tensorflow/model/ImageNet.scala renamed to src/main/scala/tensorflow/model/InceptionV3.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ import java.nio.file.{Files, Paths}
44

55
import scala.collection.JavaConverters._
66

7-
class ImageNet(graphPath: String, humanLabelPath: String, labelMapPath: String) extends TensorFlowModel with Labelable {
7+
class InceptionV3(graphPath: String, humanLabelPath: String, labelMapPath: String) extends TensorFlowModel with Labelable {
88

99
def this(modelPath: String) = this(
1010
s"$modelPath/classify_image_graph_def.pb",

0 commit comments

Comments
 (0)