trainmodeltrafficsignrecognize

Train model Nhận diện biển báo giao thông sử dụng deep learning

Yêu cầu:

Python:

• python 3.6.9

Packages:
*Khuyến khích sử dụng Anaconda 3 tạo một environment mới tên "opencv" để cài tất cả packages như hình

• pandas 0.25.2
• numpy 1.17.2
• matplotlib 3.1.1
• tensorflow 2.0.0
• opencv 3.4.2
• scikit-image 0.15.0
• scikit-learn 0.21.3
• pillow 6.2.1

Chuẩn bị input:

1. Tải bộ ảnh input tại đây
2. Copy file Train.zip vào trong thư mục input
3. Giải nén

Run project:

Activate biến môi trường Anaconda 3

source ospath/anaconda3/anaconda3/bin/activate

Activate môi trường chứa các packages cần thiết

conda activate opencv

Di chuyển đến thư mục chứa project

cd parentProjectPath/train_model_traffic_sign_recognize-master 

Chạy file main.py trong thư mục main

python main/main.py

Sau khi chạy xong chúng ta sẽ được file model.h5 trong thư mục model

Lưu ý

• Source code tham khảo bài viết tại link: Nhận diện biển báo giao thông

• Một số cấu hình trong file main.py

  • inputDirPath : Đường dẫn đến thư mục chứa input
  • modelDirPath : Đường dẫn đến thư mục lưu model
  • modelName : Tên model sẽ lưu
  • epochs : Số lần train model
  • classes : Số lớp input = Số tập ảnh train = Số thư mục trong thư mục Train