Skip to content

kaylode/trash-segmentation

Repository files navigation

Real-Time Trash Instaces Segmentation

Project for studying

Introduction

alt text alt text

  • Cấu trúc của Yolact (ảnh lấy từ paper) alt text
  • Yolact quan trọng vấn đề tốc độ hơn chính xác (độ chính xác giảm không nhiều, nhưng tốc độ vượt trội hơn so với các model khác)

Referenced Papers:

  1. YOLACT: Real-time Instance Segmentation

  2. YOLACT++: Better Real-time Instance Segmentation (2019)

Datasets:

Pretrained Weights

Model Image Size Backbone Dataset Weights
YOLACT 550 x 550 Resnet101-FPN COCO yolact_base_54_800000.pth
YOLACT 550 x 550 Resnet101-FPN TACO yolact_base_10_16500.pth

Backbone Resnet101: Link

Referenced Codes:

Guide Video:

  • COCO Annotation Format Guide Watch the video

Hướng dẫn cài đặt và sử dụng trên colab

  • Vào link này, thêm lối tắt drive của dataset và annotiation về drive của mình:

  • Thêm lối tắt 3 file pretrained weights ở trên về drive của mình

  • Tải file colab và chạy thử

Hướng dẫn thêm dataset mới

  • Đầu tiên clone source code của Yolact từ trang github của họ về
  • Vào thư mục yolact/data mở file config.py và làm giống như ví dụ dưới đây

alt text alt text

  • Sau đó bạn push folder yolact đó lên github cá nhân, và cũng upload dataset mới lên drive cá nhân
  • Tham khảo colab của mình, thay vào đó link github và drive của bạn

Report

alt text

About

Project for studying

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published