search

royorel / FFHQ-Aging-Dataset

FFHQ-Aging Dataset
Other
260 stars 25 forks source link

image and crop size in data_loader #14

Open zhihongp opened 2 years ago

zhihongp commented 2 years ago

A bit confused by the data loading process. It seems the downloaded data is 256x256, but applied with bilinear resizing to 513x513 in data_loader.py. Is the resizing needed as the deeplab model is trained on 513x513 (or 512?) data?

royorel commented 2 years ago

@zhihongp You are correct, the deeplab network was trained on 513x513 images. (It was supposed to be 512x512, but we had a typo :) )

The data itself can be generated at any desired resolution, (see the readme file) but the segmentation network works on 513x513. The output segmentation maps are then resized with nearest neighbor interpolation.