towhee.trainer.utils.plot_utils.image_folder_statistic

towhee.trainer.utils.plot_utils.image_folder_statistic(root: str, classes: Optional[List[str]] = None, show_bar: Optional[bool] = False)

Show count statistic infos for torchvision.datasets.ImageFolder. Put another way, the image root path structure just need to be like this: ``` root-|

#classname1#-|

image-1.jpg image-2.jpg

#classname2-|

image-1.jpg image-2.jpg

#classnameN-|

image-1.jpg image-2.jpg

``` :param root: Root of ImageFolder instance. :type root: str :param classes: The specified classes of statistic. If not None, all class statistic will be return. :type classes: Optional[List[str]] :param show_bar: Whether showing bar graph. :type show_bar: Optional[bool]

Returns:

(dict)

Count statistic info, where key is category name, value is the count.