from keras.models import load_model
import numpy as np
import cv2
model = load_model("Xception_tune_86.h5")
liebiao=['公交站','公园','客厅','楼梯','马路','盲道','人行横道','卫生间','卧室','走廊']
predict_dir = '5.jpg'
frame2=cv2.imread(predict_dir)
pre_x = []
input1 = cv2.cvtColor(frame2, cv2.COLOR_BGR2RGB)
pre_x.append(input1) # input一张图片
pre_x = np.array(pre_x) / 255.0
pre_y = model.predict(pre_x)
print(pre_y[0])
yuce=max(pre_y[0])
for i in range(10):
if yuce==pre_y[0][i]:
diji=i
print(liebiao[diji])
