kivenyangming / LabelImgSam Public

Notifications You must be signed in to change notification settings
Fork 2
Star 6

这是一个通过mobileSam推理获取预测框进行筛选出目标框的脚本，协助标注人员快速标注工作，在小目标标注上具有明显优势

6 stars 2 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 12 Commits
mobile_sam		mobile_sam
scripts		scripts
weights		weights
Config.py		Config.py
README.md		README.md
car.jpg		car.jpg
main.py		main.py

Repository files navigation

LabelImgSam

这是一个通过mobileSam推理获取预测框进行筛选出目标框的脚本，协助标注人员快速标注工作，在小目标标注上具有明显优势

流程：

CV2读取图像并转RGB格式
调用mobileSam网络获取mask_generator
筛选出图像推理结果中的area和bbox数组
根据area数组中数据由小到大进行排序并将bbox数组内元素顺序同area顺序
将bbox中的xywh格式数据更改为xyxy格式
将xyxy数组转换为SortedDict格式数据
两点法获取目标框最小外接矩形

示意图

About

这是一个通过mobileSam推理获取预测框进行筛选出目标框的脚本，协助标注人员快速标注工作，在小目标标注上具有明显优势

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%