37 lines
895 B
Python
37 lines
895 B
Python
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)
|