DIRE
DIRE copied to clipboard
Real Class PNG Format Image Request
I've observed that the images in the "real" folder are in JPG format, while the generated images are in PNG format. I wrote the following code to convert the images in the "adm" folder into JPG:
import os
import cv2
import glob
from pathos.multiprocessing import ProcessingPool as Pool
from tqdm import tqdm
png_images = glob.glob("./*.png")
png_images.sort()
def png2jpg(png_image):
img = cv2.imread(png_image)
os.remove(png_image)
cv2.imwrite(png_image[:-4] + ".jpg", img)
with Pool(8) as p:
list(tqdm(p.imap(png2jpg, png_images), total=len(png_images)))
and after that i ran the test.py
, and the results of lsun_bedroom/lsun_bedroom
using lsun_adm.pth
:
ACC: 0.49500
AP: 0.50112
R_ACC: 0.79800
F_ACC: 0.19200
Before converting the images, the results were as follows (as seen in issue #9)
ACC: 0.89900
AP: 0.99890
R_ACC: 0.79800
F_ACC: 1.00000
Can you provide real class PNG dire image? Thanks.