Custom category-conditioned localization dataset