ImageAI là project phát hiện & nhận diện vật thể trong ảnh. Sử dụng thuật toán thị giác máy tính và deep learning cho ra kết quả chính xác cao. Project viết bằng Python và dễ dàng chạy trên nhiều hệ điều hành khác nhau.
Bài viết này tương đối đơn giản và dễ dàng thực hành. Chỉ cần sử dụng vài câu lệnh là các bạn có thể test thử độ chính xác trước khi nghiên cứu sâu hơn.
Để sử dụng, bạn cần phải biết cách cài đặt các thư viện Python & hiểu cách sử dụng VirtualEnv.
Cách sử dụng
Đầu tiên các bạn download Python, mình sử dụng Python 3.6.5 x64, lưu ý bắt buộc là Python x64 vì TensorFlow sử dụng Python x64.
Tạo 1 folder để chứa các file trong bài viết này, ví dụ: D:\HelloImageAI
Sử dụng VirtualEnv tạo thư mục venv trong folder vừa tạo để cài đặt lib vào đó. Không sử dụng VirtualEnv cũng được nhưng nếu bạn chạy nhiều lib trong 1 máy thì dễ xung đột.
cd D:\HelloImageAI
virtualenv venv
venv\Scripts\activate
Các lib sử dụng cho ImageAI nếu sử dụng version mới nhất sẽ báo lỗi tùm lum. Do đó mình đã test thử và tìm ra được các version chạy tốt với nhau không sinh ra lỗi.
-
Cài đặt Tensorflow
pip3 install tensorflow==1.4.0
Cài đặt Keras
pip3 install keras==2.1.0
Cài đặt Numpy
pip3 install numpy==1.17.2
Cài đặt Pillow
pip3 install pillow==7.0.0
Cài đặt scipy
pip3 install scipy==1.4.1
Cài đặt h5py để đọc file model
pip3 install h5py==2.10.0
Cài đặt matplotlib
pip3 install matplotlib==3.1.2
Cài đặt opencv-python
pip3 install opencv-python==4.1.1.26
Cuối cùng là cài đặt ImageAI
pip3 install imageai==2.1.0
Vào link chứa example của ImageAI, download file object_detection.py. Trong đó có tên file là 2.jpg các bạn sửa lại cho phù hợp.
Sau khi chạy kết quả xuất ra console
Một ví dụ khác
