Custom-train Ultralytics YOLO11 để ước tính tư thế chó

Nếu tư thế của chó có thể cung cấp cho bạn thông tin về cảm xúc của chúng thì sao? Việc theo dõi chúng thủ công suốt ngày đêm không hề dễ dàng. Tuy nhiên, nhờ những tiến bộ trong trí tuệ nhân tạo (AI) và thị giác máy tính, chúng ta có thể phân tích video theo thời gian thực để hiểu rõ hơn hành vi của chúng.

Cụ thể, các model thị giác máy tính như Ultralytics YOLO11 có thể giúp ước tính tư thế và theo dõi chuyển động của chó, cung cấp những thông tin giá trị về tình trạng của chúng. Điều này hoạt động như thế nào? Khả năng ước tính tư thế của YOLO11 có thể xác định các điểm chính trên cơ thể đối tượng để hiểu tư thế và chuyển động của chúng.

Ngoài ra, YOLO11 có thể được huấn luyện tùy chỉnh trên một dataset được thiết kế cho việc ước tính tư thế chó, giúp phân tích chính xác ngôn ngữ cơ thể của thú cưng. Gói Python của Ultralytics hỗ trợ Dog-Pose Dataset, giúp việc huấn luyện và triển khai các model AI thị giác cho chó trở nên dễ dàng hơn. Công nghệ này là một phần của thị trường pet tech đang phát triển mạnh, được định giá 9,4 tỷ USD vào năm 2024 và dự kiến đạt 64 tỷ USD vào năm 2037.

Nguồn cảm hứng cho bài viết này là Blues, Dog Executive Officer (DEO) của chúng tôi. Nếu xem trang Giới thiệu, bạn sẽ thấy Blues là một thành viên được coi trọng trong đội ngũ và đóng vai trò quan trọng trong việc duy trì bầu không khí vui vẻ tại Ultralytics!

Trong bài viết này, chúng ta sẽ tìm hiểu cách huấn luyện tùy chỉnh Ultralytics YOLO11 bằng Dog-Pose Dataset để ước tính tư thế chó. Chúng ta cũng sẽ khám phá các ứng dụng thực tiễn của model trong chăm sóc thú cưng và phân tích hành vi.

Hình 1. Làm quen với Blues, Dog Executive Officer (DEO) tại Ultralytics.

Tổng quan về Dog-Pose Dataset#

Dataset là tập hợp dữ liệu được sử dụng để huấn luyện các model machine learning. Đối với ước tính tư thế, một dataset lý tưởng bao gồm các hình ảnh có gán nhãn các điểm chính để biểu diễn vị trí cơ thể. Dataset cũng nên có nhiều tư thế, góc chụp, điều kiện ánh sáng và nền khác nhau để giúp model học cách nhận dạng và dự đoán tư thế chính xác. Sự đa dạng này giúp model đáng tin cậy hơn khi sử dụng trong thực tế.

Dog-Pose Dataset, được Ultralytics hỗ trợ, được thiết kế riêng để giúp các model học và nhận dạng tư thế chó hiệu quả. Dataset này bao gồm hơn 8.400 hình ảnh được chú thích thuộc nhiều giống chó khác nhau, với nhãn chi tiết cho 24 điểm chính, chẳng hạn như đuôi, tai và bàn chân.

Hình 2. Tổng quan về Dog-Pose Dataset.

Cách huấn luyện tùy chỉnh Ultralytics YOLO11 với Dog-Pose Dataset#

Huấn luyện tùy chỉnh YOLO11 với Dog-Pose Dataset là một quy trình đơn giản. Để bắt đầu, bạn cần thiết lập môi trường bằng cách cài đặt gói Python của Ultralytics, trong đó bao gồm tất cả công cụ cần thiết cho việc huấn luyện và đánh giá.

Ultralytics tích hợp sẵn hỗ trợ cho Dog-Pose Dataset, loại bỏ nhu cầu gán nhãn thủ công và cho phép bạn bắt đầu huấn luyện ngay. Sau khi thiết lập xong, bạn có thể huấn luyện YOLO11 trên Dog-Pose Dataset chỉ bằng vài dòng code, như minh họa trong hình bên dưới.

Trong quá trình huấn luyện, model học cách phát hiện và theo dõi tư thế chó thuộc các giống, điều kiện ánh sáng và môi trường khác nhau. Sau khi huấn luyện, bạn có thể trực quan hóa kết quả và fine-tune model để cải thiện độ chính xác và hiệu năng.

Hình 3. Đoạn code minh họa việc huấn luyện tùy chỉnh Ultralytics YOLO11 trên Dog-Pose Dataset.

Nếu gặp vấn đề trong quá trình huấn luyện model, dưới đây là một số mẹo khắc phục sự cố giúp bạn nhanh chóng giải quyết vấn đề:

• Kiểm tra kết nối Internet: Dog-Pose Dataset được tự động tải xuống khi bạn chạy script huấn luyện. Hãy đảm bảo kết nối Internet ổn định để tránh lỗi tải xuống.
• Cập nhật Ultralytics: Đảm bảo bạn đang sử dụng phiên bản mới nhất của gói Python Ultralytics.
• Kiểm tra lỗi trong console: Đọc kỹ mọi thông báo lỗi, vì chúng thường cung cấp manh mối về những gì cần được khắc phục.

Bạn có thể xem Hướng dẫn xử lý các vấn đề thường gặp của Ultralytics để biết thêm mẹo khắc phục sự cố.

Điều gì xảy ra trong quá trình huấn luyện model tùy chỉnh?#

Bạn có thể thắc mắc điều gì xảy ra phía sau khi huấn luyện tùy chỉnh Ultralytics YOLO11 trên Dog-Pose Dataset. Hãy cùng xem xét kỹ hơn quy trình này.

Thay vì bắt đầu từ đầu, chúng ta sử dụng model Ultralytics YOLO11-pose đã được huấn luyện trước, vốn đã được huấn luyện trên COCO-Pose dataset. Model được huấn luyện trước này có thể phát hiện các điểm chính của con người vì COCO-Pose được thiết kế cho việc ước tính tư thế người. Trên thực tế, không cần huấn luyện bổ sung, bạn có thể sử dụng YOLO11-Pose để thực hiện ước tính tư thế người ngay lập tức.

Thông qua transfer learning, chúng ta điều chỉnh model này specifically cho việc ước tính tư thế chó, giúp model nhận dạng các điểm chính như chân, đuôi và đầu. Bằng cách cho model tiếp xúc với các ví dụ cụ thể về chó, model học cách tập trung vào những đặc điểm thiết yếu này.

Trong quá trình huấn luyện, một số phần của model không thay đổi, giữ lại kiến thức tổng quát thu được từ COCO dataset. Các phần khác được huấn luyện lại để cải thiện độ chính xác khi ước tính tư thế chó. Model học bằng cách so sánh các dự đoán với các điểm chính thực tế trong dataset và điều chỉnh để giảm lỗi. Theo thời gian, quy trình này giúp model theo dõi chuyển động của chó chính xác hơn.

Transfer learning cũng giúp model thích ứng với các giống, kích thước và kiểu chuyển động khác nhau, đảm bảo model hoạt động đáng tin cậy trong các tình huống thực tế.

Hình 4. Các điểm chính của Blues và em gái Happy được phát hiện.

Ưu điểm của Ultralytics YOLO11 trong việc ước tính tư thế chó#

Có rất nhiều model thị giác máy tính, vậy điều gì khiến YOLO11 trở thành lựa chọn phù hợp cho việc ước tính tư thế chó?

Ultralytics YOLO11 nổi bật nhờ tốc độ và độ chính xác theo thời gian thực, trở thành một lựa chọn tuyệt vời cho việc ước tính tư thế chó. Model hoạt động tốt hơn các phiên bản trước cả về độ chính xác lẫn tốc độ. Với ít hơn 22% số parameter so với YOLOv8, model đạt độ chính xác trung bình (mAP) cao hơn trên COCO dataset, nghĩa là phát hiện đối tượng chính xác và hiệu quả hơn. Tốc độ xử lý nhanh khiến model đặc biệt phù hợp với các ứng dụng thời gian thực, nơi khả năng phát hiện nhanh và đáng tin cậy là yếu tố thiết yếu.

Ngoài ước tính tư thế, Ultralytics YOLO11 còn hỗ trợ các tác vụ thị giác máy tính như instance segmentation và object tracking, giúp xây dựng giải pháp AI thị giác toàn diện hơn để theo dõi chó của bạn. Những tính năng này có thể cải thiện việc theo dõi chuyển động, phân tích hành vi và chăm sóc thú cưng nói chung.

Hình 5. Ultralytics YOLO11 hoạt động: dễ dàng phân đoạn Blues!

Các ứng dụng của việc ước tính tư thế chó và Ultralytics YOLO11#

Tiếp theo, hãy cùng thảo luận về các ứng dụng thực tế của việc ước tính tư thế chó và tác động của công nghệ này đối với việc chăm sóc thú cưng.

Cải thiện huấn luyện thú cưng bằng ước tính tư thế chó#

Ước tính tư thế chó có thể giúp việc huấn luyện chó trở nên thông minh và hiệu quả hơn. Giả sử một camera đang được sử dụng để ghi lại chuyển động của chó; đây là lúc Ultralytics YOLO11 có thể phát huy tác dụng. Model có thể phát hiện các điểm chính như chân, đuôi và đầu, sau đó phân tích chúng để nhận dạng các hành động như ngồi, đứng yên hoặc nằm xuống.

Nếu chó không thực hiện đúng hành động, hệ thống có thể cung cấp phản hồi tức thì thông qua một ứng dụng, hỗ trợ người huấn luyện theo thời gian thực. Điều này giúp việc huấn luyện hiệu quả, chính xác và phản hồi nhanh hơn theo tiến bộ của chó.

Ví dụ, hãy xem xét việc dạy chó ngồi theo hiệu lệnh. Hệ thống có thể theo dõi tư thế của chó và phát hiện xem chó đã ngồi hoàn toàn hay chưa. Nếu chó hạ thấp cơ thể nhưng chưa ngồi hẳn, hệ thống có thể phát hiện hành động chưa hoàn chỉnh và gửi phản hồi tức thì qua ứng dụng. Người huấn luyện sẽ được thông báo để thực hiện những điều chỉnh nhỏ trong quá trình huấn luyện, chẳng hạn như củng cố hiệu lệnh hoặc hướng dẫn chó vào đúng tư thế.

Hình 6. Blues hỗ trợ đội ngũ thực hiện ước tính tư thế chó bằng Ultralytics YOLO11

Nâng cao chăm sóc thú y bằng các model ước tính tư thế#

Thị giác máy tính có thể chuyển đổi cách bác sĩ thú y tiếp cận việc chăm sóc động vật. Khả năng phân tích chính xác các chi tiết của ước tính tư thế chó giúp dễ dàng phát hiện các kiểu chuyển động bất thường và xác định những vấn đề sức khỏe tiềm ẩn.

Ví dụ, bác sĩ thú y theo dõi một chú chó đang hồi phục sau chấn thương dây chằng có thể sử dụng Ultralytics YOLO11, được huấn luyện trên Dog-Pose Dataset, để phân tích tự động. Tình trạng đi khập khiễng hoặc thay đổi vị trí chân có thể dễ dàng được phát hiện. Việc theo dõi liên tục 24/7 được hỗ trợ bởi thị giác máy tính cung cấp thông tin rõ ràng về quá trình hồi phục của chó, giúp bác sĩ xác định liệu phương pháp điều trị có hiệu quả hay cần điều chỉnh.

Tương lai của việc ước tính tư thế chó#

Khi công nghệ tiếp tục phát triển, các giải pháp như ước tính tư thế chó bằng YOLO11 nhiều khả năng sẽ đóng vai trò lớn hơn trong theo dõi động vật và chăm sóc sức khỏe. Trên thực tế, YOLO11 có thể được tích hợp với công nghệ đeo, chẳng hạn như vòng cổ thông minh và thiết bị theo dõi sức khỏe, để theo dõi các chỉ số sức khỏe chính như nhịp tim, mức độ hoạt động và kiểu vận động.

Ví dụ, vòng cổ thông minh được trang bị cảm biến chuyển động có thể theo dõi dáng đi hoặc chạy của chó, trong khi khả năng ước tính tư thế của Ultralytics YOLO11 phân tích tư thế theo thời gian thực. Nếu hệ thống phát hiện chuyển động bất thường, chẳng hạn như đi khập khiễng hoặc cứng khớp, hệ thống có thể đối chiếu dữ liệu này với nhịp tim và mức độ hoạt động để đánh giá cảm giác khó chịu hoặc chấn thương tiềm ẩn. Chủ nuôi và bác sĩ thú y sử dụng những thông tin này để xác định vấn đề sớm và chủ động xử lý.

Với những tiến bộ này, ước tính tư thế chó đang phát triển vượt ra ngoài việc theo dõi chuyển động đơn giản - công nghệ này đang trở thành một phần quan trọng của hệ thống chăm sóc thú cưng toàn diện do AI điều khiển, giúp chó khỏe mạnh hơn, an toàn hơn và được theo dõi tốt hơn theo thời gian thực.

Những điểm chính#

Với những đổi mới như Ultralytics YOLO11 và Dog-Pose Dataset, chúng ta đang mở ra những khả năng mới trong thị giác máy tính. Những tiến bộ này giúp chúng ta hiểu rõ hơn về hành vi và sức khỏe của chó theo những cách trước đây chưa thể thực hiện.

Bằng cách theo dõi chính xác tư thế chó, chúng ta có thể cải thiện việc huấn luyện, theo dõi sức khỏe và nâng cao hiệu quả chăm sóc thú cưng. Dù trong nghiên cứu, chăm sóc thú y hay huấn luyện chó, AI thị giác đang tạo ra những cách thức thông minh hơn để chăm sóc chó và cải thiện tình trạng của chúng.

Tham gia cộng đồng của chúng tôi và khám phá repository GitHub của chúng tôi để tìm hiểu thêm về AI. Khám phá cách những đổi mới như thị giác máy tính trong chăm sóc sức khỏe và AI trong xe tự lái đang định hình tương lai. Xem các tùy chọn cấp phép của chúng tôi để bắt đầu các dự án AI thị giác ngay hôm nay.