使用Python實現簡易的數據標注工具

以增加工作效率為目的，最近一直在著手構建一個AI ToolBox

這兩天，我為其中的預處理工具目錄添加了數據標注模塊，本文所介紹內容的代碼見這里

該數據標注模塊包含以下幾個demo

gui_tkinter_exercise.py

data_annotation_tool.py

image_select_tool.py

一個簡單的GUI練習

gui_tkinter_exercise.py 是一個簡單的 GUI 練習demo，它展示了python的GUI庫Tkinter的最基本使用方法

如果你對Tkinter不熟悉，閱讀gui_tkinter_exercise.py的代碼是很不錯的選擇，你將了解如何創建一個GUI窗體，如何創建Button，如何關聯事件，以及創建一個數據標注工具的必要知識。

GUI demo的界面如下所示：

數據標注工具

data_annotation_tool.py 是一個數據標注demo，展示了如何實現一個數據標注工具

如何使用?

首先，文本框中展示了數據標注結果的保存路徑（目前為了方便，是寫死在代碼中的）

然后，點擊‘start’按鈕來選擇待標注數據所在文件夾

這之后，指定文件夾中的所有圖片都會被一個接一個按順序的加載

點擊類別按鈕（例如‘Dog’和‘Cat’）來對一張圖片進行標注

點擊‘NumericUpDown’按鈕來跳到特定的圖片開始標注（可用于繼續之前的工作）

點擊 'QUIT’來退出

數據標注工具的運行截圖如下所示：

數據篩選工具

image_select_tool.py 是一個數據篩選demo，展示了如何實現一個數據篩選工具。這個工具正是由我工作的一個實際需求而誕生的

潛在應用場景：

假設我們有很多組圖像，我們想要從每組中篩選出一張或多張圖片。當你有這樣類似的需求時，這個工具就適合幫助你提高工作效率

程序的運行邏輯是：

首先，文本框中展示了數據標注結果的保存路徑（目前為了方便，是寫死在代碼中的）

然后，點擊‘start’按鈕來選擇待篩選數據所在根文件夾

隨后，所有根文件夾下的子文件夾都會被一個接一個的加載

每一次，單個子文件夾下的所有圖像都會被展示在canvas畫布中

使用鼠標點擊來篩選出想要的圖像，它會被自動保存到指定路徑下

使用鼠標滾輪來進行圖片的快速瀏覽

按下空格來切換到下一組圖像

該工具的使用截圖如下所示：

總結

以上是生活随笔為你收集整理的使用Python实现简易的数据标注工具的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。