前一段時間看到交流群中有人需要關鍵點標注軟件,但是筆者並沒有找到特別適合的軟件,所以想自己寫一個簡單的軟件。筆者代碼借鑒了BBox-Label-Tool的代碼,將檢測部分的代碼改成了關鍵點的代碼。
列舉一下這個軟件的優點:
- 代碼量小,只有400多行,只需要掌握基礎的tkinter就很容易看懂。
- 占用空間小,打包成exe后只有10M。
- 可以定制化,可以自定義標注格式,直接導出你期望的標注類型,只需要改很少的代碼。
- 界面功能,添加新功能情況下可以自己快速開發,進行界面設計,開發新功能。
以下軟件是用tkinter開發的一個關鍵點標注工具,代碼量相比於pyqt比較小,功能和界面上可能略微不如pyqt強大、方便,但是優點是簡單,只需要python不需要其他額外的庫文件。
軟件的界面如下圖所示(筆者自己設計的):
使用教程視頻:
視頻教程也上傳到bilibili了: https://www.bilibili.com/video/BV145411t7Vp
使用說明
快捷鍵:
a: 上一張圖片
d: 下一張圖片
s: 保存當前圖片
保存格式:
關鍵點個數
每行一個關鍵點坐標,歸一化到0-1
同類軟件:
目標框: https://github.com/puzzledqs/BBox-Label-Tool
先目標框再關鍵點: https://github.com/Jeff-sjtu/labelKeypoint
代碼參考
https://github.com/puzzledqs/BBox-Label-Tool
anchor-free的出現,代表了用關鍵點定位成為檢測的一種新的參考點。對於一些比較小的目標,比如眼睛、鼻子定位,不需要用bounding box來,只需要用關鍵點定位即可。筆者根據這個軟件標注了100張圖片作為數據集,並寫了一個簡單的關鍵點識別的demo,以后有機會出一篇文章講解。