Hướng dẫn đánh nhãn dữ liệu Instance segmentation cho YOLO v8 bằng tool YOLO polygon trên Windows

YOLO Polygon là phần mềm dùng để đánh nhãn vật thể để training segmentation.

Mục lục

1. Giới thiệu
2. Chức năng của chương trình
3. Cách vẽ polygon
4. Công cụ crop polygon
5. Download

1. Giới thiệu

Với bài toán phát hiện đối tượng nằm cạnh nhau, có hình dạng không phải đa giác lồi nên có thể chồng chập lên nhau thì phải vẽ đường gấp khúc bao lấy đối tượng. Công cụ YOLO Polygon giúp các bạn đánh nhãn dễ dàng trên Windows.

Đây là chương trình Windows Form viết bằng ngôn ngữ C#, sử dụng .NET framework nên dễ dàng chạy trên Windows 10 và 11.

2. Chức năng của chương trình

Để training YOLO segmentation, bạn cần có dataset gồm ảnh và label, cùng với đó là file class name (.names) chứa tên các class cần đánh nhãn.

Ví dụ như file plate.names có 2 class tương ứng với ID là 0 và 1.
plate_long
plate_short

1. Vẽ hình đa giác lên ảnh và save với format của YOLO
2. Auto save: hiện chữ màu xanh lá thông báo đã save khi vẽ xong
3. Crop polygon để kiểm tra đánh nhãn đúng hay sai

Format của file text có dạng:

<class-index> <x1> <y1> <x2> <y2> ... <xn> <yn>

<class-index> là ID của class name
x1, y1,… là các điểm với giá trị là tỷ lệ của điểm trong khoảng [0:1]

3. Cách vẽ polygon

Add/edit polygon

Vẽ polygon mới (cách 1)

• Double click vào 1 point bất kỳ để tạo polygon mới
• Click vào point tiếp theo và lần lượt bao hết vị trí vật thể
• Click vào point đầu tiên sẽ hoàn tất việc tạo mới 1 polygon

Vẽ polygon mới (cách 2)

• Drag & drop trên ảnh, chương trình sẽ vẽ hình chữ nhật
• Click vào các point để chọn, sau đó có thể di chuyển point
• Click vào point đầu tiên sẽ hoàn tất việc tạo mới 1 polygon

Di chuyển point

• Click vào point để chọn, khi đó point có màu đỏ và con chuột có hình mũi tên 4 hướng
• Sau đó drag point đến vị trí mới

Thêm point vào polygon

• Di chuột đến đường thẳng nối 2 point, khi đó con chuột có hình chữ thập
• Double click để tạo 1 point mới

4. Công cụ crop polygon

Click vào menu Crop Polygon để hiện popup, trong menu này các bạn chọn các folder ảnh, label và class file tương ứng. Sau đó ấn nút Crop để crop polygon vào folder.

Vật thể sẽ được crop vào folder tương ứng, các bạn có thể kiểm tra xem có sai class hoặc sai kích thước hay không.

5. Download