Làm chủ việc triển khai YOLOv5 trên Google Cloud Platform (GCP) Deep Learning VM#
Bắt đầu hành trình trí tuệ nhân tạo (AI) và học máy (ML) có thể rất hào hứng, đặc biệt khi bạn tận dụng sức mạnh và tính linh hoạt của một nền tảng điện toán đám mây. Google Cloud Platform (GCP) cung cấp các công cụ mạnh mẽ được thiết kế riêng cho những người đam mê cũng như chuyên gia ML. Một trong những công cụ đó là Deep Learning VM, được cấu hình sẵn cho các tác vụ khoa học dữ liệu và ML. Trong hướng dẫn này, chúng ta sẽ tìm hiểu quy trình thiết lập Ultralytics YOLOv5 trên GCP Deep Learning VM. Cho dù bạn đang bước những bước đầu tiên trong ML hay là một kỹ sư dày dặn kinh nghiệm, hướng dẫn này cung cấp một lộ trình rõ ràng để triển khai các mô hình phát hiện đối tượng được hỗ trợ bởi YOLOv5.
🆓 Ngoài ra, nếu bạn là người dùng GCP mới, bạn sẽ gặp may mắn với ưu đãi $300 tín dụng miễn phí để khởi động các dự án của mình.
Ngoài GCP, hãy khám phá các tùy chọn bắt đầu nhanh có sẵn khác cho YOLOv5, chẳng hạn như Google Colab Notebook 
để có trải nghiệm dựa trên trình duyệt, hoặc khả năng mở rộng của Amazon AWS. Hơn nữa, những người yêu thích container có thể sử dụng hình ảnh Docker chính thức của chúng tôi có sẵn trên Docker Hub 
Bước 1: Tạo và Cấu hình Deep Learning VM của bạn#
Hãy bắt đầu bằng cách tạo một máy ảo được tối ưu hóa cho học sâu:
- Điều hướng đến GCP marketplace và chọn Deep Learning VM.
- Chọn một instance n1-standard-8; nó mang lại sự cân bằng với 8 vCPU và 30 GB bộ nhớ, phù hợp cho nhiều tác vụ ML.
- Chọn một GPU. Sự lựa chọn phụ thuộc vào khối lượng công việc của bạn; thậm chí một GPU T4 cơ bản cũng sẽ tăng tốc đáng kể việc huấn luyện mô hình.
- Chọn hộp kiểm 'Install NVIDIA GPU driver automatically on first startup?' để có trải nghiệm thiết lập liền mạch.
- Phân bổ 300 GB SSD Persistent Disk để tránh nghẽn cổ chai I/O.
- Nhấp vào 'Deploy' và để GCP cung cấp Deep Learning VM tùy chỉnh của bạn.
VM này đi kèm với các công cụ và framework thiết yếu đã được cài đặt sẵn, bao gồm bản phân phối Python Anaconda, gói gọn một cách thuận tiện nhiều phần phụ thuộc cần thiết cho YOLOv5.
Bước 2: Chuẩn bị VM cho YOLOv5#
Sau khi thiết lập môi trường, hãy tiến hành cài đặt và chuẩn bị YOLOv5:
# Clone the YOLOv5 repository
git clone https://github.com/ultralytics/yolov5
cd yolov5
# Install dependencies
pip install -r requirements.txtQuá trình thiết lập này đảm bảo bạn có phiên bản môi trường Python 3.8.0 trở lên và PyTorch 1.8 hoặc mới hơn. Các tập lệnh của chúng tôi tự động tải xuống các mô hình và tập dữ liệu từ bản phát hành YOLOv5 mới nhất, đơn giản hóa quy trình bắt đầu huấn luyện mô hình.
Bước 3: Huấn luyện và Triển khai các Mô hình YOLOv5 của bạn#
Sau khi hoàn tất thiết lập, bạn đã sẵn sàng để huấn luyện, đánh giá, dự đoán và xuất với YOLOv5 trên GCP VM của bạn:
# Train a YOLOv5 model on your dataset (e.g., yolov5s)
python train.py --data coco128.yaml --weights yolov5s.pt --img 640
# Validate the trained model to check Precision, Recall, and mAP
python val.py --weights yolov5s.pt --data coco128.yaml
# Run inference using the trained model on images or videos
python detect.py --weights yolov5s.pt --source path/to/your/images_or_videos
# Export the trained model to various formats like ONNX, CoreML, TFLite for deployment
python export.py --weights yolov5s.pt --include onnx coreml tfliteChỉ với vài câu lệnh, YOLOv5 cho phép bạn huấn luyện các mô hình phát hiện đối tượng tùy chỉnh phù hợp với nhu cầu cụ thể của bạn hoặc sử dụng trọng số được huấn luyện trước để có kết quả nhanh chóng trên các tác vụ khác nhau. Khám phá các tùy chọn triển khai mô hình khác sau khi xuất.
Phân bổ Swap Space (Tùy chọn)#
Nếu bạn đang làm việc với các tập dữ liệu đặc biệt lớn có thể vượt quá RAM của VM, hãy cân nhắc thêm không gian swap để tránh lỗi bộ nhớ:
# Allocate a 64GB swap file
sudo fallocate -l 64G /swapfile
# Set the correct permissions for the swap file
sudo chmod 600 /swapfile
# Set up the Linux swap area
sudo mkswap /swapfile
# Enable the swap file
sudo swapon /swapfile
# Verify the swap space allocation (should show increased swap memory)
free -hHuấn luyện trên Tập dữ liệu Tùy chỉnh#
Để huấn luyện YOLOv5 trên tập dữ liệu tùy chỉnh của bạn trong GCP, hãy làm theo các bước chung sau:
-
Chuẩn bị tập dữ liệu của bạn theo định dạng YOLOv5 (hình ảnh và tệp nhãn tương ứng). Xem tổng quan về tập dữ liệu của chúng tôi để được hướng dẫn.
-
Tải lên tập dữ liệu của bạn lên GCP VM bằng
gcloud compute scphoặc tính năng SSH của bảng điều khiển web. -
Tạo một tệp cấu hình tập dữ liệu YAML (
custom_dataset.yaml) chỉ định các đường dẫn đến dữ liệu huấn luyện và đánh giá của bạn, số lượng lớp và tên các lớp. -
Bắt đầu quá trình huấn luyện bằng cách sử dụng YAML tập dữ liệu tùy chỉnh của bạn và có thể bắt đầu từ trọng số được huấn luyện trước:
# Example: Train YOLOv5s on a custom dataset for 100 epochs python train.py --img 640 --batch 16 --epochs 100 --data custom_dataset.yaml --weights yolov5s.pt
Để có hướng dẫn toàn diện về cách chuẩn bị dữ liệu và huấn luyện với các tập dữ liệu tùy chỉnh, hãy tham khảo tài liệu Huấn luyện Ultralytics YOLOv5.
Tận dụng Lưu trữ Đám mây (Cloud Storage)#
Để quản lý dữ liệu hiệu quả, đặc biệt là với các tập dữ liệu lớn hoặc nhiều thí nghiệm, hãy tích hợp quy trình làm việc YOLOv5 của bạn với Google Cloud Storage:
# Ensure Google Cloud SDK is installed and initialized
# If not installed: curl https://sdk.cloud.google.com/ | bash
# Then initialize: gcloud init
# Example: Copy your dataset from a GCS bucket to your VM
gsutil cp -r gs://your-data-bucket/my_dataset ./datasets/
# Example: Copy trained model weights from your VM to a GCS bucket
gsutil cp -r ./runs/train/exp/weights gs://your-models-bucket/yolov5_custom_weights/Phương pháp này cho phép bạn lưu trữ các tập dữ liệu lớn và các mô hình đã huấn luyện một cách an toàn và tiết kiệm chi phí trên đám mây, giảm thiểu yêu cầu lưu trữ trên instance VM của bạn.
Kết luận#
Xin chúc mừng! Giờ đây bạn đã được trang bị để khai thác các khả năng của Ultralytics YOLOv5 kết hợp với sức mạnh tính toán của Google Cloud Platform. Thiết lập này mang lại khả năng mở rộng, hiệu quả và tính linh hoạt cho các dự án phát hiện đối tượng của bạn. Cho dù là để khám phá cá nhân, nghiên cứu học thuật hay xây dựng các giải pháp công nghiệp, bạn đã thực hiện một bước tiến quan trọng vào thế giới AI và học máy trên đám mây.
Hãy cân nhắc sử dụng Ultralytics Platform để có trải nghiệm hợp lý, không cần mã để huấn luyện và quản lý các mô hình của bạn.
Hãy nhớ ghi lại tiến trình của bạn, chia sẻ thông tin chi tiết với cộng đồng Ultralytics sôi động và tận dụng các tài nguyên như thảo luận trên GitHub để cộng tác và hỗ trợ. Bây giờ, hãy tiến lên và đổi mới với YOLOv5 và GCP!
Bạn muốn tiếp tục nâng cao các kỹ năng học máy của mình? Hãy đi sâu vào tài liệu của chúng tôi và khám phá Blog Ultralytics để biết thêm các hướng dẫn và thông tin chi tiết. Hãy để cuộc phiêu lưu AI của bạn tiếp tục!