Commit 368b32d

committed

replaced elif with else. added missing imports

1 parent b481adc commit 368b32dCopy full SHA for 368b32d

File tree

-3

lines changed

-3

lines changed

Lines changed: 5 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -2,8 +2,11 @@`
`2`	`2`	`import argparse`
`3`	`3`	`import os`
`4`	`4`	`import tensorflow as tf`
`5`		`-import matplotlib.pyplot as plt`
	`5`	`+from PIL import Image`
	`6`	`+from io import BytesIO`
`6`	`7`	`import pathlib`
	`8`	`+import glob`
	`9`	`+import matplotlib.pyplot as plt`
`7`	`10`
`8`	`11`	`from object_detection.utils import ops as utils_ops`
`9`	`12`	`from object_detection.utils import label_map_util`
`@@ -21,7 +24,6 @@ def load_model(model_path):`
`21`	`24`	`return model`
`22`	`25`
`23`	`26`
`24`		`-`
`25`	`27`	`def load_image_into_numpy_array(path):`
`26`	`28`	`"""Load an image from file into a numpy array.`
`27`	`29`
`@@ -96,7 +98,7 @@ def run_inference(model, category_index, image_path):`
`96`	`98`	`line_thickness=8)`
`97`	`99`	`plt.imshow(image_np)`
`98`	`100`	`plt.show()`
`99`		`- elif`
	`101`	`+ else:`
`100`	`102`	`image_np = load_image_into_numpy_array(image_path)`
`101`	`103`	`# Actual detection.`
`102`	`104`	`output_dict = run_inference_for_single_image(model, image_np)`

Comments

(0)