better validation total time output

main
vlpr 6 months ago
parent 3121584af4
commit dca830bbe0

@ -5,6 +5,7 @@ from common.utils import PSNR, cal_ssim, logger_info, _rgb2ycbcr, modcrop
from pathlib import Path
from PIL import Image
import time
from datetime import timedelta
def val_image_pair(model, hr_image, lr_image, output_image_path=None, device='cuda'):
with torch.inference_mode():
@ -70,7 +71,7 @@ def valid_steps(model, datasets, config, log_prefix=""):
len(test_dataset),
np.mean(lr_areas),
total_area,
total_time
timedelta(seconds=total_time)
]
results.append(row)
column_names = [
@ -82,7 +83,7 @@ def valid_steps(model, datasets, config, log_prefix=""):
'Image count',
'AVG image area',
'Total area',
'Total time, s'
'Total time'
]
config.logger.info("\n" + str(pd.DataFrame([row], columns=column_names).set_index('Dataset').T))
config.writer.flush()

Loading…
Cancel
Save