Cách xử lý ảnh đầu vào để train theo batch như thế nào ? about vietocr HOT 4 CLOSED

Hi e, thư viện này là train theo từng batch e nhé.
Để train theo từng batch thì đơn giản là e chỉ cần lấy những ảnh có kích thước bằng nhau gộp thành một batch rồi forward qua thôi.
Đơn giản là e sẽ tính chiều dài của mỗi ảnh, sắp sếp trước các ảnh có cùng width thì cho vào cùng một bucket rồi sao đó sampling từ các bucket.

Em có thể tham khảo file này vietocr/loader/DataLoader.py

from vietocr.

from vietocr.

e có thể resize width về bucket gần nhất mà e quy định nhé.
ví dụ e có 6 bucket 50, 60, 70, 80, 90, 100 thì với một ảnh cho trước e sẽ resize width về bucket về gần để giảm biến dạng ảnh.
anh không xử dụng padding

from vietocr.

from vietocr.