[update] modify comment for multi models

This commit is contained in:
NatLee 2025-01-07 04:05:11 +08:00
parent e668cd415e
commit 85e2d8d863

View File

@ -28,7 +28,7 @@ class Demography(ABC):
And switch to batch prediction if receives batched images. And switch to batch prediction if receives batched images.
Args: Args:
img_batch: Batch of images as np.ndarray (n, 224, 224, 3), with n >= 1. img_batch: Batch of images as np.ndarray (n, x, y, c), with n >= 1, x = image width, y = image height, c = channel
""" """
if not self.model_name: # Check if called from derived class if not self.model_name: # Check if called from derived class
raise NotImplementedError("virtual method must not be called directly") raise NotImplementedError("virtual method must not be called directly")