不是用我的名字！推斷擴散模型使用的輸入字串的藝術家姓名

這是官方儲存庫

不是用我的名字！推斷 Diffusion Models 使用的輸入字串的藝術家名稱。羅伯托·萊奧塔、奧利佛·朱迪斯、盧卡·瓜內拉、塞巴斯蒂亞諾·巴蒂亞托。 ICIAP 2023。

不是用我的名字！

安裝

使用 conda 或 miniconda 輕鬆安裝所需的依賴項。

安裝 conda 後，執行以下命令：

 # 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

然後按照資料集部分和檢查點部分中的說明下載所需的檔案。

數據集

至少下載以下資料集之一：

數據集	# 原始影像	# AI生成的圖像	關聯	國標
小的	2350	2350	關聯	2.7
中等的	4130	8519	即將推出	/

下載資料集後，將其解壓縮到resources夾中。資料夾結構應如下所示。

檢查站

下載以下檢查點：連結並將其放置在resources/ckpts資料夾中。資料夾結構應如下所示。

推理

注意：要執行推理，您需要下載檢查點和資料集（至少是小資料集）。

對單一影像進行推理

若要對單一影像運行推理，請執行以下命令：

 # 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 >

結果將保存在--results-folder <path-to-results-folder>中。

對單一影像進行推理 - 範例

 # 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

推理用法

 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

引文

如果您發現此程式碼對您的研究有用，請引用我們的論文：

 @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}
}