[update] add hint for the shape of input img

This commit is contained in:
NatLee 2025-01-07 04:54:35 +08:00
parent ba0d0c5bb6
commit 29141b3cd5

View File

@ -54,7 +54,7 @@ class ApparentAgeClient(Demography):
# Preprocessing input image or image list. # Preprocessing input image or image list.
imgs = self._preprocess_batch_or_single_input(img) imgs = self._preprocess_batch_or_single_input(img)
# Prediction # Prediction from 3 channels image
age_predictions = self._predict_internal(imgs) age_predictions = self._predict_internal(imgs)
# Calculate apparent ages # Calculate apparent ages