GoodWin/Deep-Multi-scale
0
1U
2 �9_� � @ sF d dl Z d dlZd dlmZ d dlZd dlZd dlZG dd� d�ZdS )� N)�utilc @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d3� Zd S ) � BaseOptionsc C s4 d| _ d S )NF)� initialized)�self� r �;/home/lxm/ExpCodes/GitDFDNet/DFDNet/options/base_options.py�__init__5 s zBaseOptions.__init__c C s |j dtddd� |j dtddd� |j dtd d6d� |j d td d
d� |j dtddd� |j dtddd� |j ddtdd� |j dtddd� |j dtd d d� |j d d d d!� |j d"td#d$d� |j d%td&d'd� |j d(td)d*d� |j d+d d,d!� |j d-d.td/d� d0| _|S )1Nz --batchSize� zinput batch size)�type�default�helpz--ngf�@ z$# of gen filters in first conv layerz --gpu_ids�0z+gpu ids: e.g. 0 0,1,2, 0,2. use -1 for CPUz--nameZfacefh_dictionaryzDname of the experiment. It decides where to store samples and modelsz--modelZfaceDictz4chooses which model to use. cycle_gan, pix2pix, testz--which_directionZBtoAz AtoB or BtoAz7--nThreads� z# threads for loading data)r r8 r z--checkpoints_dirz
./checkpointszmodels are saved herez--norm�instancez-instance normalization or batch normalizationz--serial_batchesZ9store_truezMif true, takes images in order to make batches, otherwise takes them randomly)�actionr z--resize_or_cropZ degradationzcscaling and cropping of images at load time [resize_and_crop|crop|scale_width|scale_width_and_crop]z --init_typeZkaimingz9network initialization [normal|xavier|kaiming|orthogonal]z --init_gaing{�G�z�?z1scaling factor for normal, xavier and orthogonal.z --verbosez.if specified, print more debugging informationz--suffix� z`customized suffix: opt.name = opt.name + suffix: e.g., {model}_{which_model_netG}_size{loadSize}T)� add_argument�int�str�floatr )r �parserr r r �10initialize
s" zBaseOptions.initializec C s| | j s tjtjd�}| �|�}|�� \}}|j}t�|�}||| j �}|�� \}}|j11}t �|�}||| j �}|| _ |�
� S )N)Zformatter_class)r �argparseZArgumentParserZ ArgumentDefaultsHelpFormatterr Zparse_known_args�model�modelsZget_option_setter�isTrainZ dataset_mode�datar Z12parse_args)r r �opt�_Z13model_nameZmodel_option_setterZ dataset_nameZdataset_option_setterr r r �gather_options! s �14 15 16 zBaseOptions.gather_optionsc 17 C s� d}|d7 }t t|��� �D ]F\}}d}| j�|�}||krHdt|� }|d�t|�t|�|�7 }q |d7 }t|� tj �18|j |j �}t
�|� tj �19|d�}t|d��} | �|� | �d� W 5 Q R X d S ) Nr z*----------------- Options ---------------20z [default: %s]z{:>25}: {:<30}{}21z)----------------- End -------------------zopt.txtZwt�22)�sorted�vars�itemsr Z get_defaultr �format�print�os�path�joinZcheckpoints_dir�namer Zmkdirs�open�write)23r r �message�k�vZcommentr Zexpr_dir� file_nameZopt_filer r r �
print_options< s 24 25zBaseOptions.print_optionsc C s� | � � }| j|_|jrF|jdkr6d|jjf t|�� nd}|j| |_|j�d�}g |_|D ] }t|�}|dkr\|j� |� q\t26|j�dkr�t j �
|jd � || _| jS )Nr r �,r )r r �suffixr% r# r* �gpu_ids�splitr �append�len�torchZcudaZ27set_devicer )r r r3 Zstr_idsZstr_id�idr r r �parseP s $ zBaseOptions.parseN)�__name__�28__module__� __qualname__r r r r1 r: r r r r r s29 r )r r' r r8 r r r r r r r �<module> s