a  þd· ã@sjddlZddlZddlmZddlmZddlmZddlm Z ddd „Z e  ¡dd d „ƒZ dd d„Z dS)éN)Úlinalg)Útqdm)Ú InceptionV3ÚcudaTFcCs(tdg||d}t |¡ ¡ |¡}|S)Né)Ú resize_inputÚnormalize_input)rÚnnZ DataParallelÚevalÚto)ÚdevicerrÚ inception©rú\/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/basicsr/metrics/fid.pyÚload_patched_inception_v3 srcCsˆ|durt|ddd}nd}g}|D]F}|r6| d¡| |¡}||ƒd |jdd¡}| | d¡¡q$|rx| ¡t |d¡}|S) a^Extract inception features. Args: data_generator (generator): A data generator. inception (nn.Module): Inception model. len_generator (int): Length of the data_generator to show the progressbar. Default: None. device (str): Device. Default: cuda. Returns: Tensor: Extracted features. NÚbatchZExtract)ÚtotalÚunitÚdescéréÿÿÿÿÚcpu) rÚupdater ÚviewÚshapeÚappendÚcloseÚtorchÚcat)Zdata_generatorr Z len_generatorr ZpbarÚfeaturesÚdataÚfeaturerrrÚextract_inception_featuress   r"çíµ ÷Æ°>c Cs|j|jksJdƒ‚|j|jks(Jdƒ‚tj||dd\}}t |¡ ¡s~tdƒt |jd¡|}t ||||¡}t |¡rÆtj t  |¡j dddsÀt  t  |j ¡¡}td |›ƒ‚|j}||} | | } t |¡t |¡d t |¡} | | } | S) aÞNumpy implementation of the Frechet Distance. The Frechet distance between two multivariate Gaussians X_1 ~ N(mu_1, C_1) and X_2 ~ N(mu_2, C_2) is d^2 = ||mu_1 - mu_2||^2 + Tr(C_1 + C_2 - 2*sqrt(C_1*C_2)). Stable version by Dougal J. Sutherland. Args: mu1 (np.array): The sample mean over activations. sigma1 (np.array): The covariance matrix over activations for generated samples. mu2 (np.array): The sample mean over activations, precalculated on an representative data set. sigma2 (np.array): The covariance matrix over activations, precalculated on an representative data set. Returns: float: The Frechet Distance. z'Two mean vectors have different lengthsz)Two covariances have different dimensionsF)ZdispzNProduct of cov matrices is singular. Adding {eps} to diagonal of cov estimatesrgü©ñÒMbP?)ZatolzImaginary component é)rrZsqrtmÚnpÚisfiniteÚallÚprintZeyeZ iscomplexobjZallcloseZdiagonalÚimagÚmaxÚabsÚ ValueErrorÚrealÚtrace) Zmu1Zsigma1Zmu2Zsigma2ZepsZcov_sqrtÚ_ÚoffsetÚmZ mean_diffZ mean_normr.ZfidrrrÚ calculate_fid2s" "r2)rTF)Nr)r#)Únumpyr%rZtorch.nnr ZscipyrrZbasicsr.archs.inceptionrrZno_gradr"r2rrrrÚs