Tidak dengan namaku! Menyimpulkan nama artis dari string masukan yang digunakan oleh Model Difusi

Ini adalah gudang resmi

Tidak dengan namaku! Menyimpulkan nama artis dari string masukan yang digunakan oleh Diffusion Models . Roberto Leotta , Oliver Giudice , Luca Guarnera , Sebastiano Battiato . ICIAP 2023.

Tidak dengan namaku!

Instalasi

Gunakan conda atau miniconda untuk menginstal dependensi yang diperlukan dengan mudah.

Dengan conda terinstal, jalankan perintah berikut:

 # 1. Clone the repository
git clone https://github.com/ictlab-unict/not-with-my-name.git
# 2. navigate to the repository folder
cd not-with-my-name
# 3. create the environment
conda env create -f docs/environment.yml

Kemudian ikuti instruksi di bagian dataset dan bagian checkpoints untuk mendownload file yang diperlukan.

Kumpulan data

Unduh setidaknya salah satu dari kumpulan data berikut:

Kumpulan data	#Gambar asli	# AI menghasilkan gambar	Link	GB
Kecil	2350	2350	link	2.7
Sedang	4130	8519	segera hadir	/

Setelah mendownload dataset, ekstrak di folder resources . Struktur folder harus seperti berikut.

Pos pemeriksaan

Unduh checkpoint berikut: link dan letakkan di folder resources/ckpts . Struktur folder harus seperti berikut.

Kesimpulan

NB : untuk menjalankan inferensi Anda perlu mendownload checkpoint dan dataset (setidaknya yang kecil).

Inferensi pada satu gambar

Untuk menjalankan inferensi pada satu gambar, jalankan perintah berikut:

 # 1. activate the environment
conda activate not-w-my-name-env
# 2. run inference
python src/inference.py --dataset-folder < path-to-dataset > --query-img < path-to-query-image > --model-ckpt < path-to-checkpoint > --cuda --results-folder < path-to-results-folder >

Hasil akan disimpan di --results-folder <path-to-results-folder> .

Inferensi pada satu gambar - contoh

 # 1. activate the environment
conda activate not-w-my-name-env
# 2. run inference
python src/inference.py --dataset-folder resources/small-dataset/ --query-img resources/small-dataset/pablo_picasso/ai_generated/102_0.png --model-ckpt resources/ckpts/siamese_not_w_my_name.ckpt --cuda --results-folder results

Penggunaan inferensi

 usage: inference.py [-h] [--show-time] [--debug] 
                    --dataset-folder DATASET_FOLDER --results-folder RESULTS_FOLDER 
                    --query-img QUERY_IMAGE [--distance-th DISTANCE_TH] [--cuda]
                    --model-ckpt MODEL_CKPT

Not with my name inference by Roberto Leotta

optional arguments:
  -h, --help            show this help message and exit
  --show-time           show processing time
  --debug               flag for development debugging
  --dataset-folder DATASET_FOLDER
                        dataset folder path
  --results-folder RESULTS_FOLDER
                        results folder path
  --query-img QUERY_IMAGE
                        query image path
  --distance-th DISTANCE_TH
                        distance threshold for the query image. Default: 0.5
  --cuda                use CUDA for inference
  --model-ckpt MODEL_CKPT
                        siamese model checkpoint

Kutipan

Jika Anda merasa kode ini berguna untuk penelitian Anda, silakan kutip makalah kami:

 @inproceedings{leotta2023not,
   title={Not with my name! Inferring artists' names of input strings employed by Diffusion Models},
   author={Leotta, Roberto and Giudice, Oliver and Guarnera, Luca and Battiato, Sebastiano},
   booktitle={International Conference on Image Analysis and Processing},
   year={2023},
   organization = {Springer}
}