David310/Detect_AI-generated_Image
4
1U
2 �ajf� � @ s> d dl Z d dlZdd� Zdd� Zdddgd d3d gfd d
�ZdS )� Nc C s4 t | t�r(t | t�s(| D ] }t|� qnt| � d S �N)�4isinstance�list�str�mkdir)�paths�path� r �,/root/autodl-tmp/UniversalFakeDetect/util.py�mkdirs s r c C s t j�| �st �| � d S r )�osr �exists�makedirs)r r r r5 r
s r g6ףp=7�?g�v��/�?g�C�l���?g Zd;�O�?gy�&1��?g�������?c C s8 | t �|�d d d �d d f t �|�d d d �d d f S r )�torch�Tensor)Ztens�mean�stdr r r8 � unnormalize s �� �r )r r r r r r r r r9 �<module> s