「硬核」labelme 图片中显示标签
這是一篇小眾的博客,大部分小伙伴可能使用不到,但是對需要的小伙伴絕對是一劑良藥!希望對有需要的小伙伴有用!
labelme 雖然主要用于數(shù)據(jù)的標(biāo)注,但是標(biāo)注完的數(shù)據(jù)集在檢查的時候,檢查標(biāo)注框?qū)?yīng)的標(biāo)簽是否正確,此工作隨不常用,但是用到時卻十分不方便,每張圖片都需要一個一個框的點擊,然后查看對應(yīng)的標(biāo)簽是否正確,十分耗時耗力。
在此,博主在labelme的基礎(chǔ)上,將對應(yīng)的標(biāo)簽一并顯示出來,此舉使得不再需要一一點擊查看,而是直接掃描一眼即可,十分方便,至少可以提高 10倍
修改好的代碼以上傳至 viatorsun.blog.csdn.net
以防萬一,建議 小伙伴只將其中的 app.py 、widgets/canvas.py 、widgets/label_dialog.py 文件替換,不建議替換 __main__.py
共需要修改兩個文件,具體操作如下:
1、app.py
在 def loadFile(self, filename=None): 函數(shù)中,大概1533行代碼處添加 以下兩行代碼
2、widgets/canvas.py
腳本導(dǎo)入 PyQt5的幾個包from PyQt5.QtGui import QPainter,QFont,QColor ,在def __init__(self, *args, **kwargs): 中添加 屬性 self._shapes = [] 用于記錄標(biāo)注信息,最后在 def paintEvent(self, event): 添加如下信息
3、新增加標(biāo)注情況的顯示
4、修改 Edit label 對話框尺寸
在widgets/label_dialog.py 文件中,48行 [super(LabelDialog, self).__init__(parent)] 之后新增以下語句即可,小伙伴可根據(jù)自己情況修改寬高比
from qtpy.QtCore import QSizeself.resize(QSize(300, 500))至此即可
問題答疑
項目相對簡單,可能存在潛藏Bug,還望見諒!!!
總結(jié)
以上是生活随笔為你收集整理的「硬核」labelme 图片中显示标签的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 【配电网重构】基于粒子群求解配电网重构模
- 下一篇: Spring Boot学习之旅:(六)s