-
Notifications
You must be signed in to change notification settings - Fork 0
/
predict.py
29 lines (23 loc) · 860 Bytes
/
predict.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
from keras.models import load_model
import argparse
import script.utils
def parse_args():
desc = "Vehicle Classification"
parser = argparse.ArgumentParser(description=desc)
parser.add_argument('--model', type=str, default='model/vehicle_model_saved.h5', help='Where Is Model File?')
parser.add_argument('--img', type=str, default='data/1.jpg', help='What Is Images Path?')
return parser.parse_args()
def main():
args = parse_args()
if args is None:
exit()
# Load Model
model = load_model(args.model)
# Convert Image To Numpy Array
image = script.utils.load_image(args.img)
# Predict Image Based On Model
label = model.predict(image)
# Print Result
print("Predicted Class (0 - Cars , 1- Planes): ", round(label[0][0], 2))
if __name__ == '__main__':
main()