David310/Detect_AI-generated_Image
4
1U
2 ��^f � @ s2 d dl Z d dlZd dlZd dlZG dd� d�ZdS )� Nc @ s6 e Zd Zdd� Zdd� Zdd� Zdd� Zd
d3d �Zd S )� BaseOptionsc C s4 d| _ d S )NF)� initialized)�self� r �</root/autodl-tmp/UniversalFakeDetect/options/base_options.py�__init__ s zBaseOptions.__init__c C s� |j ddd� |j dtddd� |j dd d5� |j d d d� |j d
tdd� |j ddd� |j dtdd� |j ddd� |j ddd� |j dd dd� |j dd dd� |j d d d d� |j d d d d� |j d!d"d#d� |j d$td%d&d� |j d'd d6� |j d(td)d*d� |j d+td)d,d� |j d-td.d/d� |j d0td1d2d� |j d3td4d5d� |j d6d7td8d9� |j d:td;d<d� |j d=d d>d?� |j d@tdAdBd� |j dCd dDd?� |j dEtdFdGd� |j dHtdIdJd� |j dKdLtdMd9� dN| _|S )ONz--mode�binary)�defaultz--archZres50zsee my_models/__init__.py)�typer �helpz--fix_backbone�7store_true)�actionz --rz_interp�bilinearz --blur_probg �?)r8 r z9--blur_sigz0.0,3.0z10--jpg_probz --jpg_methodzcv2,pilz11--jpg_qualz30,100z--real_list_pathzponly used if data_mode==ours: path for the list of real images, which should contain train.pickle and val.pickle)r r z--fake_list_pathzponly used if data_mode==ours: path for the list of fake images, which should contain train.pickle and val.picklez--wang2020_data_pathzIonly used if data_mode==wang2020 it should contain train and test foldersz --data_mode�ourszwang2020 or oursz --data_label�trainz3label to decide whether train or validation datasetz--weight_decayg zloss weight for l2 regz --class_balz --batch_size� zinput batch sizez12--loadSizezscale images to this sizez13--cropSize�� zthen crop to this sizez --gpu_ids�0z+gpu ids: e.g. 0 0,1,2, 0,2. use -1 for CPUz--nameZexperiment_namezDname of the experiment. It decides where to store samples and modelsz
--num_threads� z# threads for loading data)r r14 r z--checkpoints_dirz
./checkpointszmodels are saved herez--serial_batcheszMif true, takes images in order to make batches, otherwise takes them randomly)r
r z--resize_or_cropZscale_and_cropzhscaling and cropping of images at load time [resize_and_crop|crop|scale_width|scale_width_and_crop|none]z --no_flipz:if specified, do not flip the images for data augmentationz --init_type�normalz9network initialization [normal|xavier|kaiming|orthogonal]z --init_gaing{�G�z�?z1scaling factor for normal, xavier and orthogonal.z--suffix� zTcustomized suffix: opt.name = opt.name + suffix: e.g., {model}_{netG}_size{loadSize}T)� add_argument�str�float�intr )r �parserr r r �15initialize s> zBaseOptions.initializec C s8 | j s tjtjd�}| �|�}|�� \}}|| _|�� S )N)�formatter_class)r �argparse�ArgumentParser� ArgumentDefaultsHelpFormatterr �parse_known_argsr �16parse_args)r r �opt�_r r r �gather_options1 s �17 zBaseOptions.gather_optionsc 18 C s� d}|d7 }t t|��� �D ]F\}}d}| j�|�}||krHdt|� }|d�t|�t|�|�7 }q |d7 }t|� tj �19|j |j �}t
�|� tj �20|d�}t|d��} | �|� | �d� W 5 Q R X d S ) Nr z*----------------- Options ---------------21z [default: %s]z{:>25}: {:<30}{}22z)----------------- End -------------------zopt.txt�wt�23)�sorted�vars�itemsr � get_defaultr �format�print�os�path�join�checkpoints_dir�name�utilZmkdirs�open�write)24r r# �message�k�v�commentr Zexpr_dir� file_nameZopt_filer r r �
print_options> s 25 26zBaseOptions.print_optionsTc C sN | � � }| j|_|jrF|jdkr6d|jjf t|�� nd}|j| |_|rT| �|� |j�d�}g |_|D ] }t |�}|dkrj|j�27|� qjt |j�dkr�t j
�|jd � |j�d�|_dd� |j�d�D �|_|j�d�|_dd� |j�d�D �|_t |j�dk�r*tt|jd |jd d ��|_nt |j�dk�rBtd28��|| _| jS ) Nr r$ �,r c S s g | ] }t |��qS r )r ��.0�sr r r �29<listcomp>l s z%BaseOptions.parse.<locals>.<listcomp>c S s g | ] }t |��qS r )r r= r r r r@ n s � � z1Shouldn't have more than 2 values for --jpg_qual.)r% �isTrain�suffixr, r) r2 r; �gpu_ids�splitr �append�len�torch�cuda�30set_device� rz_interp�blur_sig�31jpg_method�jpg_qual�list�range�32ValueErrorr# )r r; r# rD Zstr_idsZstr_id�idr r r �parseR s2 $ 33 "zBaseOptions.parseN)T)�__name__�34__module__� __qualname__r r r% r; rT r r r r r s35 &
r )r r. r3 rI r r r r r �<module> s