基于yolov8训练推理脚本pyinstaller打包openvino报错问题

最新推荐文章于 2024-10-12 16:41:10 发布

原创

最新推荐文章于 2024-10-12 16:41:10 发布 · 1.4k 阅读

26 ·

CC 4.0 BY-SA版权

文章标签：

#YOLO #openvino #人工智能

一、正常打包自定义脚本

本文基于ultralyics的yolov8和halcon的标注hdict文件，写了一个一键生成yolo数据集、训练、推理、模型导出、测试的脚本。并且将该脚本编译成traintest.pyd文件（需要可私信），使用主函数yoloit.p文件调用上面traintest.pyd动态库模块，进行训练推理等一系列操作。其中pyd编译过程可以参考：

py编译pyd

最终生成如下两个文件，其中yoloit.py为终端输入参数argparse

import argparse
import traintest

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Yoloit arguments')
    parser.add_argument('--nomkdata', action='store_false', default=True, help='if make yolodataset')
    parser.add_argument('--notrain', action='store_false', default=True, help='if train  model')
    parser.add_argument('--noexport', action='store_false', default=True, help='if exported model')
    parser.add_argument('--notest', action='store_false', default=True, help='if test model')
    parser.add_argument('--resume', action='store_true', default=False, help='if resume model')
    parser.add_argument('--alldir', type=str, default=None, help='Directory for all files')
    parser.add_argument('--jsonpath', type=str,
                        default=r'D:\ALLvision\pycharmproject\halcon2yolo\printer\data\printe