ไม่ใช่ชื่อของฉัน! การอนุมานชื่อศิลปินของสตริงอินพุตที่ใช้โดย Diffusion Models

นี่คือพื้นที่เก็บข้อมูลอย่างเป็นทางการของ

ไม่ใช่ชื่อของฉัน! การอนุมานชื่อศิลปินของสตริงอินพุตที่ใช้โดย Diffusion Models โรแบร์โต ลีโอตต้า , โอลิเวอร์ จูดิซ , ลูก้า กัวเนรา , เซบาสเตียน บัตติอา โต้ ไอซีไอเอพี 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}
}