Train model on customize dataset

[ledinhtri97]

Hello, Minh có một bài toán tương tự nhưng repo gốc không có code train model. mình có thể dùng repo nay train trên tập dữ liệu tuỳ chỉnh được không? thank you! Mình xin thông tin để trao đổi và nếu có thể cùng đóng góp code cho nó hoạt động với nhé

[HVQuan02]

Bạn có thể tham khảo train script của mình hoặc của https://github.com/v-vietlq/thesis (nhóm này có điều chỉnh lại model gốc)

[ledinhtri97]

@HVQuan02 mình xem bình luận của bạn so sánh bạn train được mAP 30% so với 90% bên repo gốc. repo này mô hình bạn train có cải thiện mAP hơn lúc đó không?

[HVQuan02]

Repo này mình vẫn dùng model gốc của tác giả và hiệu quả thấp 30%, nên mình chuyển sang dùng model của nhóm kia thì tăng gần 70%, đây là link repo nếu bạn cần https://github.com/HVQuan02/PETA_Thesis_With_K18

[ledinhtri97]

@HVQuan02 mình dùng tập dữ liệu cá nhân. bạn cho mình thông tin cấu trúc dữ liệu thư mục sắp xếp hay cần những gì cho giống tập dữ liệu bạn dùng cho repo trên với nhé. Mình có request data từ web mà chưa có nhận được thông tin

[HVQuan02]

dataset gồm thư mục images chứa 1883 folder albums, 2 file json (đính kèm bên dưới) là event_type.json chứa label event của từng album và image_importance.json chứa importance scores của từng image trong mỗi album. Ngoài ra có 1 thư mục độc lập split_dir chứa train_split (tên các album dùng để train) và val_split (tên các album dùng để val) event_type.json image_importance.json

[ledinhtri97]

@HVQuan02 mình xin data sample qua email này nhé? triledinh102097@gmail.com data folder chỉ vẫn vài mẫu tượng trưng thôi, chứ không cần hết đâu. Cám ơn bạn trước nhé

[HVQuan02]

repo https://github.com/HVQuan02/PETA_Thesis_With_K18 của mình vừa đạt mAP 80% (so với 90% trong paper) và spearman = 0.16 (so với 0.374 trong paper). Bạn có thể tham khảo!

[ledinhtri97]

@HVQuan02 tuyệt vời!!!. cám ơn bạn nhé. Mình nhận cũng được data rồi