iT邦幫忙

第 12 屆 iThome 鐵人賽

DAY 23
0
Software Development

【錢不夠買ps的我,只好用OpenCV來修圖了!】系列 第 23

【沒錢ps,我用OpenCV!】Day 23 - 綜合運用2,(資料標註) 用 OpenCV 來製作一個標記點小工具吧! 滑鼠與視窗控制 label points

先來看看今天的結果gif

https://ithelp.ithome.com.tw/upload/images/20201006/20120424CaI5dUn99a.png

圖片說明

-> 此篇文章的程式碼 github

Day23_標記點小工具_label_points.ipynb

前言

我們該來運用之前學過的所有東西了!
綜合運用篇就是來一次運用前面的所學!

(資料標註) 用 OpenCV 來製作一個標記點小工具吧!

我們要製作一個標記點小工具,大致要完成三個功能:

  • 主程式 (讀取圖片)
  • 控制滑鼠相關的函數 (滑鼠處理)
  • 依據滑鼠回傳的數值做出對應的反應函數 (主要功能處理)

主程式 (讀取圖片)

img_dst = cv2.imread("./testdata/cat.jpg")

print("Click on the screen and press any key for end process")
points  = get_points(img_dst)

print("\npoints list:")
print(points)

這部分就單純的讀取圖片,
我們使用我們定義的 get_points 來呼叫我們主要的功能。

控制滑鼠相關的函數 (滑鼠處理)

def mouse_handler(event, x, y, flags, data):
    if event == cv2.EVENT_LBUTTONDOWN:
        # 標記點位置
        cv2.circle(data['img'], (x,y), 3, (0,0,255), 5, 16) 

        # 改變顯示 window 的內容
        cv2.imshow("Image", data['img'])
        
        # 顯示 (x,y) 並儲存到 list中
        print("get points: (x, y) = ({}, {})".format(x, y))
        data['points'].append((x,y))

我們先定義一個控制滑鼠的函數,
OpenCV 有內建滑鼠偵測的函數 cv2.EVENT_LBUTTONDOWN
為了讓我們的功能更有回饋感,我們在點擊之處新增一個紅點,
作為我們真的有點下去的視覺上回饋,
我們在畫點之後更新畫面,即時顯示在畫面上,
並將這些點儲存起來,供等等分析使用。


依據滑鼠回傳的數值做出對應的反應函數 (主要功能處理)

def get_points(im):
    # 建立 data dict, img:存放圖片, points:存放點
    data = {}
    data['img'] = im.copy()
    data['points'] = []
    
    # 建立一個 window
    cv2.namedWindow("Image", 0)
    
    # 改變 window 成為適當圖片大小
    h, w, dim = im.shape
    print("Img height, width: ({}, {})".format(h, w))
    cv2.resizeWindow("Image", w, h)
        
    # 顯示圖片在 window 中
    cv2.imshow('Image',im)
    
    # 利用滑鼠回傳值,資料皆保存於 data dict中
    cv2.setMouseCallback("Image", mouse_handler, data)
    
    # 等待按下任意鍵,藉由 OpenCV 內建函數釋放資源
    cv2.waitKey()
    cv2.destroyAllWindows()
    
    # 回傳點 list
    return data['points']

這裡就是我們的主要功能區,
我們先建立一個 data dictionary,
將我們的圖片儲存在 data['img'] 這個欄位,
並宣告一個 data['points'] 讓我們等等來儲存點。

我們建立一個新的作業用視窗 cv2.namedWindow
並修改為符合圖片大小 cv2.resizeWindow
最後顯示圖片在視窗中 cv2.imshow

cv2.setMouseCallback("Image", mouse_handler, data)

  • 第二個參數就是我們剛剛定義滑鼠功能的函數名字,
  • 第三個參數我們傳入 data dictionary,讀取圖片並儲存點座標。

為求程式的穩定性,我們使用 OpenCV 的內建函數來釋放資源
cv2.waitKey(0)cv2.destroyAllWindows()

最後我們回傳點座標return data['points']
完成我們這次的功能。

Reference

https://zhuanlan.zhihu.com/p/143035374

https://kknews.cc/code/3oqxejy.html

https://blog.csdn.net/fanjiule/article/details/81606596

https://blog.csdn.net/yefcion/article/details/79435591


上一篇
【沒錢ps,我用OpenCV!】Day 22 - 綜合運用1,用 OpenCV 來P圖囉! 來運用各種之前學習的各種東西吧! merge two images
下一篇
【沒錢ps,我用OpenCV!】Day 24 - 綜合運用3,(應用app) 用 OpenCV 來製作一個照片文件掃描機吧! photo scanner 透視投影
系列文
【錢不夠買ps的我,只好用OpenCV來修圖了!】30

尚未有邦友留言

立即登入留言