- Download trình duyệt google chome
- Download driver google chome
- Viết chương trình lấy file ảnh từ google images
- Viết chương trình chuẩn hóa tên file ảnh
- Viết chương trình phân chia 80% ảnh train 20% ảnh test
- Cài đặt công cụ labelImg
- Tập ảnh
- Tập nhãn
- download darknet
- Đưa dữ liệu ảnh vào /darknet/data/images/
- Đưa file tọa độ rectangle vào /darknet/data/labels/
- Đưa file train.txt gồm 80% ảnh trong tập dữ liệu và0 /darknet/
- Đưa file val.txt gồm 20% ảnh trong tập dữ liệu vào /darknet/
- Tạo file yolo.data, yolo.names
- Download file darknet53.conv.74
- Chỉnh sửa số liệu trong file darknet/cfg/yolov3.cfg ,filter and class
- Chỉnh sửa file /darknet/examples/detector.c, tìm đến dòng 138
if(i%2000==0 || (i < 1000 && i%100 == 0))
- Bắt đầu train:
./darknet detector train yolo.data cfg/yolov3.cfg darknet53.conv.74
Train dữ liệu trên google colab
- Nén darknet thành darknet.zip
- Upload lên google drive
- Sử dụng [google colab]https://colab.research.google.com() để truy cập
- chọn file -> new python 3 Notebooke
- chọn runtime -> change runtime type -> đổi None sang GPU
- Kết nối đến google drive
from google.colab import drive
drive.mount('/content/drive')
!git clone https://github.com/AlexeyAB/darknet
!apt-get update
!apt-get upgrade
!apt-get install build-essential
!apt-get install cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev
!apt-get install libavcodec-dev libavformat-dev libswscale-d
!apt-get install libopencv-dev
%cd darknet
!ls
!sed -i 's/OPENCV=0/OPENCV=1/g' Makefile
!sed -i 's/GPU=0/GPU=1/g' Makefile
!ls
%cd ../
!ls
!apt install g++-5
!apt install gcc-5
!apt update
!apt upgrade
!/usr/local/cuda/bin/nvcc --version
%cd darknet
!make
!wget https://pjreddie.com/media/files/darknet53.conv.74
!rm /content/darknet/backup -r
!ln -s /content/drive/My\ Drive/ML/backup /content/darknet/backup
%cd /content/drive/My\ Drive/ML/
!unzip /content/drive/My\ Drive/ML/darknet.zip
%cd /content/darknet
!./darknet detector train data/phones.data yolo-phones.cfg darknet53.conv.74 -dont_show
- giải nén file zip
%cd /content
!unzip /content/drive/'My Drive'/ML/darknet.zip
%cd /content/darknet
- Gán quyền thực thi
!make
!chmod +x ./darknet
- Tạo liên kết mềm sang google dive
!rm /content/darknet/backup -r
!ln -s /content/drive/'My Drive'/ML/backup /content/darknet
- tiến hanh train
%cd /content/darknet
!./darknet detector train yolo.data cfg/yolov3.cfg backup/yolov3_900.weights
- download file weight mới nhất trong thư mục backup
- Lấy file weight mới nhất trong tập backup ra sung_test
- Lấy 1 tấm hình trong tập train ra test thử
- Sử dụng lệnh: python YOLO.py -i sung_test.jpg -cl yolo.names -w backup/yolov3.backup -c cfg/yolov3.cfg