AI 工具框架 7314 阅读 pytorch转为onnx并调用时遇到的问题汇总 之前一直在用yolov5训练后生成的pytorch的模型文件,后缀名为pt,也用过pth用state dict加载处理,加载和推理用的代码较多,决定转换为onnx的类型用opencv加载,减少代码量也减少第三方包的安装,这样比较适合实际应用时部署。 state dict的记录在这里:YO...... python 深度学习 YOLO opencv pytorch
工具框架 4942 阅读 Python识别图片特定区域的文字 现在我在做基于yolo的自动化测试框架,目前能识别出按钮button、输入框input等控件。但是会遇到一个情况,当前手机屏幕上或者网页上会出现多个按钮,虽然能识别出多个按钮,但是要点击哪一个呢,于是需要一个界定,首要的判断条件就是依据button上或者附近的文本。 现在用到yolo时除...... python opencv
工具框架 1868 阅读 OpenCV边界检测并获取物体边缘坐标范围 目前在用YOLO识别物体,遇到一个场景是有多个同样的物体,但要通过另一个参照物来区分。而参照物的位置和特征并不相同,于是考虑到了使用边界检测,只要在同一区域内,便可方便灵活地标识出唯一物体。 比如我用当前网站的首页来做个例子: 这里对于人眼来说,是能看到明显的区域的概念。比...... python opencv