File size: 1,470 Bytes
424919d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
# create image resolution histogram
from glob import glob
from PIL import Image
from tqdm.auto import tqdm
tmfiles_f = glob("/truemedia-eval/images/fakes/*")
tmfiles_r = glob("/truemedia-eval/images/reals/*")
files_f = glob("/home/ubuntu/Datasets/diffusiondb/train/*")
files_r = glob("/home/ubuntu/Datasets/ffhq/in-the-wild-images/train/*")
files_f += glob("/home/ubuntu/Datasets/stylegan2-ffhq/train/*")
files_r += glob("/home/ubuntu/Datasets/coco-2017-train/train2017/train/*")
Ws_f = []
Hs_f = []
for f in tqdm(files_f):
img = Image.open(f)
w, h = img.size
Ws_f.append(w)
Hs_f.append(h)
Ws_r = []
Hs_r = []
for f in tqdm(files_r):
img = Image.open(f)
w, h = img.size
Ws_r.append(w)
Hs_r.append(h)
TMWs_r = []
TMHs_r = []
for f in tqdm(tmfiles_r):
img = Image.open(f)
w, h = img.size
TMWs_r.append(w)
TMHs_r.append(h)
TMWs_f = []
TMHs_f = []
for f in tqdm(tmfiles_f):
img = Image.open(f)
w, h = img.size
TMWs_f.append(w)
TMHs_f.append(h)
import matplotlib.pyplot as plt
import numpy as np
# plot 2d (w, h)
plt.figure(figsize=(8, 10))
plt.scatter(Ws_r, Hs_r, s=3, alpha=0.5)
plt.scatter(Ws_f, Hs_f, s=3, alpha=0.5, c='g')
plt.scatter(TMWs_r, TMHs_r, s=3, alpha=0.5, c='r')
plt.scatter(TMWs_f, TMHs_f, s=3, alpha=0.5, c='y')
plt.xlabel('Width')
plt.ylabel('Height')
plt.legend(['Training-real', 'Training-fake', 'TrueMedia-real', 'TrueMedia-fake'])
# save
plt.savefig('resolution_hist.png')
|