PROJECT_DIR = os.path.abspath(os.getcwd()) USE_ADABINS = False import pathlib, shutil, os, sys from dataclasses import dataclass from functools import partial import cv2 import pandas as pd import gc import io import math import lpips from PIL import Image, ImageOps import requests from glob import glob import json from types import SimpleNamespace import torch from torch import nn from torch.nn import functional as F import torchvision import torchvision.transforms as T import torchvision.transforms.functional as TF from tqdm import tqdm from resize_right import resize from guided_diffusion.script_util import create_model_and_diffusion, model_and_diffusion_defaults from datetime import datetime import numpy as np import random import hashlib from functools import partial from numpy import asarray import time import warnings warnings.filterwarnings("ignore", category=UserWarning)