YOLOv5 🚀 trên AWS Deep Learning Instance: Hướng dẫn đầy đủ của bạn
Việc thiết lập một môi trường học sâu hiệu suất cao có thể là điều khó khăn đối với người mới bắt đầu, nhưng đừng lo lắng! 🛠️ Với hướng dẫn này, chúng tôi sẽ hướng dẫn bạn thực hiện quy trình để có được YOLOv5 đang chạy trên phiên bản AWS Deep Learning. Bằng cách tận dụng sức mạnh của Amazon Web Services (AWS), ngay cả những người mới làm quen với máy học cũng có thể bắt đầu nhanh chóng và tiết kiệm chi phí. Khả năng mở rộng của nền tảng AWS hoàn hảo cho cả thử nghiệm và triển khai sản xuất.
Các tùy chọn khởi động nhanh khác cho YOLOv5 Bao gồm của chúng tôi Máy tính xách tay Colab 
, Máy ảo học sâu GCPvà hình ảnh Docker của chúng ta tại Trung tâm Docker 
Bước 1: Đăng nhập Bảng điều khiển AWS
Bắt đầu bằng cách tạo tài khoản hoặc đăng nhập vào bảng điều khiển AWS tại https://aws.amazon.com/console/. Sau khi đăng nhập, hãy chọn dịch vụ EC2 để quản lý và thiết lập các phiên bản của bạn.
Bước 2: Khởi chạy phiên bản của bạn
Trong bảng thông tin EC2, bạn sẽ tìm thấy nút Khởi chạy phiên bản là cổng để tạo một máy chủ ảo mới.
Chọn đúng Amazon Machine Image (AMI)
Đây là nơi bạn chọn hệ điều hành và ngăn xếp phần mềm cho phiên bản của mình. Nhập ' Deep Learning ' vào trường tìm kiếm và chọn AMI Deep Learning mới nhất dựa trên Ubuntu, trừ khi nhu cầu của bạn yêu cầu khác. AMI Deep Learning của Amazon được cài đặt sẵn các khuôn khổ phổ biến và GPU trình điều khiển để đơn giản hóa quá trình thiết lập của bạn.
Chọn loại phiên bản
Đối với các tác vụ deep learning, hãy chọn một GPU Loại phiên bản thường được khuyến nghị vì nó có thể tăng tốc đáng kể việc đào tạo mô hình. Ví dụ: cân nhắc về kích thước, hãy nhớ rằng yêu cầu bộ nhớ của mô hình không bao giờ được vượt quá những gì phiên bản của bạn có thể cung cấp.
Ghi: Kích thước mô hình của bạn phải là một yếu tố trong việc chọn một phiên bản. Nếu model của bạn vượt quá RAM khả dụng của phiên bản, hãy chọn loại phiên bản khác có đủ bộ nhớ cho ứng dụng của bạn.
Để biết danh sách có sẵn GPU loại phiên bản, truy cập Loại phiên bản EC2, cụ thể là trong Điện toán tăng tốc.
Để biết thêm thông tin về GPU Giám sát và tối ưu hóa, xem GPU Giám sát và tối ưu hóa. Để biết giá, hãy xem Định giá theo nhu cầu và Định giá Spot.
Cấu hình phiên bản của bạn
Phiên bản dùng ngay Amazon EC2 cung cấp một cách thức hiệu quả về chi phí để chạy các ứng dụng vì chúng cho phép bạn đặt giá thầu cho dung lượng chưa sử dụng với một phần nhỏ chi phí tiêu chuẩn. Để có trải nghiệm liên tục giữ lại dữ liệu ngay cả khi Phiên bản Spot gặp sự cố, hãy chọn yêu cầu liên tục.
Hãy nhớ điều chỉnh phần còn lại của cài đặt phiên bản và cấu hình bảo mật nếu cần trong Bước 4-7 trước khi khởi chạy.
Bước 3: Kết nối với phiên bản của bạn
Khi phiên bản của bạn đang chạy, hãy chọn hộp kiểm và nhấp vào Kết nối để truy cập thông tin SSH. Sử dụng lệnh SSH được hiển thị trong thiết bị đầu cuối ưa thích của bạn để thiết lập kết nối với phiên bản của bạn.
Bước 4: Chạy YOLOv5
Đăng nhập vào phiên bản của bạn, bây giờ bạn đã sẵn sàng để sao chép YOLOv5 kho lưu trữ và cài đặt các phụ thuộc trong một Python 3.8 Môi trường trở lên. YOLOv5Các mô hình và bộ dữ liệu của sẽ tự động tải xuống từ bản phát hành mới nhất.
git clone https://github.com/ultralytics/yolov5 # clone repository
cd yolov5
pip install -r requirements.txt # install dependencies
Khi thiết lập môi trường, bạn có thể bắt đầu đào tạo, xác thực, thực hiện suy luận và xuất YOLOv5 Mô hình:
# Train a model on your data
python train.py
# Validate the trained model for Precision, Recall, and mAP
python val.py --weights yolov5s.pt
# Run inference using the trained model on your images or videos
python detect.py --weights yolov5s.pt --source path/to/images
# Export the trained model to other formats for deployment
python export.py --weights yolov5s.pt --include onnx coreml tflite
Các tính năng bổ sung tùy chọn
Để thêm bộ nhớ hoán đổi, có thể là vị cứu tinh cho các tập dữ liệu lớn, hãy chạy:
sudo fallocate -l 64G /swapfile # allocate 64GB swap file
sudo chmod 600 /swapfile # modify permissions
sudo mkswap /swapfile # set up a Linux swap area
sudo swapon /swapfile # activate swap file
free -h # verify swap memory
Và thế là xong! 🎉 Bạn đã tạo thành công phiên bản AWS Deep Learning và chạy YOLOv5 . Cho dù bạn mới bắt đầu với phát hiện đối tượng hay đang mở rộng quy mô để sản xuất, thiết lập này có thể giúp bạn đạt được mục tiêu học máy của mình. Chúc bạn đào tạo, xác thực và triển khai vui vẻ! Nếu bạn gặp bất kỳ trục trặc nào trong quá trình này, tài liệu AWS mạnh mẽ và hoạt động Ultralytics cộng đồng ở đây để hỗ trợ bạn.
