내 이름으로는 안돼! 확산 모델에 사용된 입력 문자열의 아티스트 이름 추론

이것은 공식 저장소입니다.

내 이름으로는 안돼! 확산 모델에 사용된 입력 문자열의 아티스트 이름을 추론합니다 . 로베르토 레오타 , 올리버 지우디스 , 루카 구아르네라 , 세바스티아노 바티아토 . 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 생성 이미지	링크	GB
작은	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}
}