Object Detection Model with Keras - MASK RCNN

Giới thiệu

Đến với computer vision, chúng ta có rất nhiều ứng dụng trong thực tế, rất nhiều trang web, blog đã trình bày cách thực hiện các hướng dẫn với các pre-train model

Ở đây mình cũng sử dụng pre-train model Mask-RCNN trong Keras để tìm vợt tenis trong hình.

Các bước thực hiện

Bước 1 : Clone Mask RCNN trên GitHub Repository :

git clone https://github.com/matterport/Mask_RCNN.git

Bước 2 : Cài đặt thư viện Mask RCNN:

cd Mask_RCNN

python setup.py install

Bước 3 : chuẩn bị dữ liệu:

chúng ta cần tải 1 số hình ảnh có vợt tenis về và gán nhãn cho chúng bằng labelimg, càng lấy nhiều hình thì kết quả càng chính xác, và tất nhiên thời gian train model sẽ càng nhiều hơn

Bước 4 : Chaỵ code và tận hưởng kết quả:

code mình để dưới này mọi người có thể copy về chạy thử

https://colab.research.google.com/drive/1o8yIdBJplxYSIIQmmYslKYyUQtUN4E7T