Tham khảo về cách xây dựng hệ thống gợi ý qua hình ảnh

Chào mọi người, em đang chuẩn bị xây dựng một hệ thống gợi ý sản phẩm bằng hình ảnh dựa trên tập dữ liệu em tìm trên Kaggle gồm 44 k ảnh và 141 class. Em đã tìm hiểu và lên ý tưởng cho hướng phát triển nhưng em không chắc là nó đúng nên em muốn lên đây để nhờ mọi người gợi ý ạ.

  1. Xây dựng pretrain model (VGG16, VGG19, RESNET50,…) bỏ lớp cuối dùng để nó cho output là vector đặc trưng của mỗi input hình ảnh.
  2. Chia tập dữ liệu hình ảnh thành 2 phần train và test.
  3. Đưa tập dữ liệu train cho model để lấy ra vector đặt chưng và lưu danh sách vào dưới dạng file
  4. Khi người dùng mỗi tìm một hình ảnh (Input), dùng model để chiết xuất vector đặt chưng của hình ảnh đó và so sánh với danh sách các vector đặt chưng của tập train bằng metric là Cosine
  5. Lấy ra k hình ảnh có metric là lớn nhất.

Đừng hỏi v, đã xác định theo AI thì là phải mày mò nhiều r. Từ kiến thức cho đến giải pháp.
Muốn biết đúng hay k thì thử thực nghiệm đi r biết. Chưa đúng chưa xịn thì lại thử cách khác.
Mà không chỉ thử không thôi, mà với mỗi hướng đi thì phải so sánh hướng đi đó với những hướng đi khác. Xem xem với hướng đi đó có thu hoạch gì mới hoặc gặp phải những rào cản nào.

Đó mới là AI, chứ thử một lần mà ra kết quả luôn thì còn gì là AI, đến việc train model cũng còn phải chạy đi chạy lại nhiều lần cơ mà. :v :V

3 Likes
83% thành viên diễn đàn không hỏi bài tập, còn bạn thì sao?