Skip to content

Commit

Permalink
change predict file
Browse files Browse the repository at this point in the history
  • Loading branch information
WZMIAOMIAO committed Feb 19, 2020
1 parent 44235b4 commit ceda35d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tensorflow_learning/Test4_goolenet/predict.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,8 @@
img = img.resize((im_width, im_height))
plt.imshow(img)

# scaling pixel value to (0-1)
img = np.array(img) / 255.
# scaling pixel value and normalize
img = ((np.array(img) / 255.) - 0.5) / 0.5

# Add the image to a batch where it's the only member.
img = (np.expand_dims(img, 0))
Expand Down

0 comments on commit ceda35d

Please sign in to comment.