halcon AI读取onnx模型并推理

*程序功能:读取onnx模型并推理
dev_update_off()
dev_close_window ()

read_dl_model ('squeezenet.onnx', DLModelHandle)
set_dl_model_param (DLModelHandle, 'type', 'classification')
get_dl_model_param(DLModelHandle, 'image_dimensions', ImageDimensions)
get_dl_model_param(DLModelHandle, 'class_names', ClassNames)
get_dl_model_param(DLModelHandle, 'class_ids', ClassIds)
set_dl_model_param (DLModelHandle, 'runtime_init', 'immediately')
dev_open_window(0, 0, 600, 400, 'black', WindowHandle)
set_font(WindowHandle, 'Microsoft YaHei UI-Bold-36')

read_image (Image, 'cat.jpg')
zoom_image_size (Image, ImagePreprocessedByte, ImageDimensions[0], ImageDimensions[1], 'constant')
convert_image_type (ImagePreprocessedByte, ImagePreprocessed, 'real')
scale_image (ImagePreprocessed, ImagePreprocessed, 1, -127)
create_dict (DLSample)
set_dict_object (ImagePreprocessed, DLSample, 'image')
apply_dl_model (DLModelHandle, DLSample, [], DLResult)
get_dict_tuple (DLResult, 'classification_confidences', Confidences)
get_dict_tuple (DLResult, 'classification_class_names', PredictClasses)
tuple_max (Confidences, Max)
tuple_find (Confidences, Max, IndexMax)
dev_clear_window ()
dev_display(Image)
Text := PredictClasses[IndexMax] + ', ' + Max$'.2f'
dev_disp_text (Text, 'window', 'top', 'left', 'red', 'box', 'false')

posted @   QuincyYi  阅读(929)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 全程不用写代码,我用AI程序员写了一个飞机大战
· DeepSeek 开源周回顾「GitHub 热点速览」
· 记一次.NET内存居高不下排查解决与启示
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了

喜欢请打赏

扫描二维码打赏

了解更多

点击右上角即可分享
微信分享提示