Python implementa programa de inferência de IA: menos de vinte linhas de código

  É muito simples implementar programas de inferência de IA em Python, exceto por referências de módulo e definições constantes, há menos de 20 linhas de programas para realizar o cálculo de inferência de IA! Existem apenas cinco funções reais para o cálculo de inferência de IA!

• A primeira etapa é usar cv.dnn.readNet () para ler o arquivo de modelo de IR no formato OpenVINO
• A segunda etapa, use net.setPreferableTarget (DEVICE) para especificar o hardware de execução do cálculo de inferência do AI
• A terceira etapa é usar cv.dnn.blobFromImage () e net.setInput () para transferir os dados da imagem para o modelo de IA
• O quarto passo é usar net.forward () para implementar o cálculo de inferência de IA
• O quinto passo é analisar os resultados do cálculo de raciocínio

O código completo para implementar o cálculo de inferência de IA com base no OpenCV e OpenVINO no Python é o seguinte

#导入opencv-openvino模块
import cv2 as cv
import time
#配置推断计算设备，IR文件路径，图片路径
DEVICE = cv.dnn.DNN_TARGET_CPU
model_xml = 'D:/tf_train/workspaces/cats_dogs/IR_model/cats_dogs_detector.xml'
model_bin = 'D:/tf_train/workspaces/cats_dogs/IR_model/cats_dogs_detector.bin'
image_file = 'D:/tf_train/workspaces/cats_dogs/images/test/3.jpg'

#读取IR模型文件
net = cv.dnn.readNet(model_xml, model_bin)

#指定AI推理计算执行硬件
net.setPreferableTarget(DEVICE)

#读取图片
img = cv.imread(image_file)

#将图片传入模型的输入张量
blob = cv.dnn.blobFromImage(img,ddepth=cv.CV_8U)
net.setInput(blob)

#执行推断计算
start = time.time()
out = net.forward()
end = time.time()
print("Infer Time:{}ms".format((end-start)*1000))

#处理推断计算结果
for detection in out.reshape(-1, 7):
    confidence = float(detection[2])
    xmin = int(detection[3] * img.shape[1])
    ymin = int(detection[4] * img.shape[0])
    xmax = int(detection[5] * img.shape[1])
    ymax = int(detection[6] * img.shape[0])

    if confidence>0.7:
        cv.rectangle(img, (xmin, ymin), (xmax, ymax), (0, 255, 0))
        conf = "{:.4f}".format(confidence)
        font = cv.FONT_HERSHEY_COMPLEX_SMALL
        cv.putText(img, conf, (xmin, ymin - 5), font, 1, (0, 0, 255))

#显示处理结果
cv.imshow("Detection results",img)
cv.waitKey(0)
cv.destroyAllWindows()
print("Inference is completed...")

Como treinar seu próprio modelo de IA e como obter arquivos de modelo OpenVINO IR?
Aprendizado avançado: "Tecnologia de reconhecimento de imagem de aprendizado profundo"