diff --git a/README.md b/README.md index be7f1d1328684d29b470c41ec54057f361421ac9..62070e8b3ff6d9b18fed46aae781cb1e4e96a2a5 100644 --- a/README.md +++ b/README.md @@ -2,8 +2,6 @@ ## Description This repository contains the materials presented in the journal paper 'Artwork recommendations guided by foundation models: survey and novel approach'. -The code of notebook 'Retrieve_combined_features.ipynb' illustrates the combination of recommendations with the same weights of genre, style and artist criteria, it can be easily modified for using with different weights. -The code for fine-tuning a classification model is based on the <a href="https://github.com/huggingface/peft/tree/main/examples/image_classification">example</a>. <a href="https://liris.cnrs.fr/page-membre/tetiana-yemelianenko">Tetiana Yemelianenko</a>, <a href="https://liris.cnrs.fr/page-membre/iuliia-tkachenko">Iuliia Tkachenko</a>, @@ -13,6 +11,8 @@ The code for fine-tuning a classification model is based on the <a href="https:/ <div style="text-align:center"><img style="margin-right: 20px" src="assets/fig8.png" alt="Recommendations" height="325" width="660"/> +The code of notebook 'Retrieve_combined_features.ipynb' illustrates the combination of recommendations with the same weights of genre, style and artist criteria, it can be easily modified for using with different weights. +The code for fine-tuning a classification model is based on the <a href="https://github.com/huggingface/peft/tree/main/examples/image_classification">example</a>. # Table of content - [Overview](#description) @@ -26,7 +26,7 @@ The code for fine-tuning a classification model is based on the <a href="https:/ The WikiArt dataset used for fine-tuning could be found <a href="https://huggingface.co/datasets/huggan/wikiart">here</a> ## Steps -To reproduce the steps first you need to finetune models for genre, style and artist classification on WikiArt dataset or used fine-tuned adapters. Then, using fine-tuned models you need to calculate embeddings for the images from WikiArt dataset and create ANNOY indexes. These files are available upon a request. +To reproduce the steps first you need to finetune models for genre, style and artist classification on WikiArt dataset or used fine-tuned adapters. Then, using fine-tuned models you need to calculate embeddings for the images from WikiArt dataset and create ANNOY indexes. Due to the relatively big size, these files are available upon a request. ## Citation ```