Gemini写的抠图工具

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

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

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

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

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

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

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

抠图效果

相关推荐
AI的探索之旅3 天前
97 个 OpenCV 实例(三十):双目立体,从标定到点云
人工智能·opencv·计算机视觉
词却3 天前
OpenCV学习:CNN 人脸检测
opencv
慧都小项3 天前
当边缘AI上产线:QtitanDocking 如何让工业 HMI 实现多视图协同
人工智能·qt·ui·边缘计算·qt6.3
yyk333243 天前
OpenCV中的疲劳检测
opencv
宿州派大星3 天前
【Python实战】:基于 OpenCV + dlib 的传统换脸术原理与实战
opencv·python3.11·dlib
陈年老古董3 天前
MediaPipe 姿态检测与脸部关键点检测
笔记·python·opencv·学习·dlib
Quz3 天前
QML SwipeDelegate 滑动委托
qt
Quz3 天前
QML 列表中的四种委托:ItemDelegate、CheckDelegate、RadioDelegate、SwitchDelegate
qt
a3535413823 天前
DFT到FFT的推导
opencv
实心儿儿3 天前
Qt — 显示类控件
qt