Gemini写的抠图工具

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

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

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

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

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

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

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

抠图效果

相关推荐
打工仔折腾 AI5 分钟前
Qt网络编程实战:UDP、TCP到HTTP的落地经验与选型
网络·人工智能·后端·python·qt·udp·ai agent 实战
Lhan.zzZ25 分钟前
Linux + Qt/QML:解决 QtQuick.VirtualKeyboard.Plugins 模块加载失败与环境变量配置
linux·数据库·qt
晨间晓28 分钟前
两张图片哪里不一样?用 OpenCV 做图像差异检测工具
人工智能·opencv·计算机视觉
richard_yuu2 小时前
OpenCV 实战第 13 篇:工程化收尾,内存拷贝、并行与一条完整流水线
人工智能·opencv·计算机视觉
晨间晓2 小时前
图片拍歪了怎么办?用华为云码道开发 OpenCV 自动纠偏工具
人工智能·opencv·华为云
誰能久伴不乏2 小时前
QT工业上位机一次「指令静默丢失」的排查
qt·docker·github
老唐7772 小时前
(附代码)使用OpenCV和关键点检测计算瓶子角度
人工智能·深度学习·opencv·目标检测·计算机视觉·ai·transformer
晨间晓4 小时前
一张图片主要是什么颜色?用 OpenCV 做图片主色提取工具
人工智能·opencv·计算机视觉
扶尔魔ocy20 小时前
【qt openGL入门笔记】OSG实现飞机模型在网格图上飞行并进行态势标绘(四)
qt·opengl·osg
橙橙笔记1 天前
QT安卓软件的制作
android·git·qt·软件