به نام خدا

استفاده از شبکه‌های از قبل آموزش داده شده با وبکم

In [2]:
from keras.applications import MobileNetV2
from keras.preprocessing import image
from keras.applications.mobilenet_v2 import preprocess_input, decode_predictions
import numpy as np
In [3]:
model = MobileNetV2(weights='imagenet')
In [4]:
def inference(x):
    x = np.expand_dims(x, axis=0)
    x = preprocess_input(x)
    preds = model.predict(x)
    return decode_predictions(preds, top=1)[0][0][1]
In [5]:
img_path = './image/elephant.jpg'
img = image.load_img(img_path, target_size=(224, 224))
x = image.img_to_array(img)
In [6]:
inference(x)
Out[6]:
'African_elephant'
In [7]:
import cv2
cap = cv2.VideoCapture(0)
while True:
    ret, frame = cap.read()
    frame = cv2.resize(frame, (224, 224))
    predicted = inference(frame[...,::-1])
    cv2.putText(frame,  predicted, (30, 30), cv2.FONT_HERSHEY_SIMPLEX, 1.0, (0, 0, 255), lineType=cv2.LINE_AA)
    cv2.imshow('Webcam', frame)
    if cv2.waitKey(1) == 13: #13 is the Enter Key
        break
cap.release()
cv2.destroyAllWindows()     
دوره مقدماتی یادگیری عمیق
علیرضا اخوان پور
23 و 30 آبان 1397
Class.Vision - AkhavanPour.ir - GitHub