Gemini写的抠图工具

工具地址(没有打包成软件):https://github.com/hou080146/RailFOD-Dataset-Generator

项目有个需求:检测某区域内有没有特定物品

问题:检测的场景有特殊性,需要通过图像增强造数据集

解决方法:通过抠图把公开数据集中的需要检测的物品抠出来,然后通过图像增强贴到场景中

花了10分钟让Gemini写了个工具,调用Facebook的SAM 2分割模型,使用opencv对物品进行抠图保存

单纯的调用opencv,没有任何UI界面和其他功能。由于我平时又使用Qt开发界面,就想着把他直接做成一个实用工具,又花了几个小时让Gemini修改成PyQt5的工具。

跟anythinglabel一样,直接在点击要抠的物品,绿色遮罩就是抠取的范围。按键S保存并下一张,空格键跳过当前图片

抠图效果

相关推荐
初阳7851 小时前
【Qt】系统相关(2)——文件
开发语言·qt
hhzz11 小时前
基于监控视频的水位尺自动识别技术方案与实现
python·opencv·yolo·图像识别·cv
尘中远15 小时前
【Qwt 7.0 系列】坐标轴与刻度系统 —— 刻度引擎、网格、图例与刻度朝内
qt·数据可视化·qcustomplot·qwt·工业软件·科学绘图
sycmancia17 小时前
Qt——多线程间的互斥
开发语言·qt
林中青木21 小时前
OpenCV 5.0 使用方法及注意事项
人工智能·opencv·计算机视觉
漂亮的摩托1 天前
基于OpenCV与SVM的车牌识别系统实现:定位、分割与分类全流程解析
opencv·支持向量机·分类
尘中远1 天前
【Qwt 7.0 系列】常用图表类型实战 —— 柱状图、散点图、箱线图与直方图
qt·qwt·工业软件·科学绘图
尘中远1 天前
【Qwt 7.0 系列】交互功能详解 —— 平移、缩放、坐标轴交互与数据拾取
qt·数据可视化·绘图·qcustomplot·qwt·科学绘图
sycmancia1 天前
Qt——进程与线程的概念
qt
郝学胜-神的一滴1 天前
Qt 高级编程 034:深耕QWidget底层内核—彻底吃透无边框窗口设计核心原理
开发语言·c++·qt·程序人生·软件开发·用户界面