def _widerface_category(): label_map = widerface_label() label_map = sorted(label_map.items(), key=lambda x: x[1]) cats = [l[0] for l in label_map] clsid2catid = {i: i for i in range(len(cats))} catid2name = {i: name for i, name in enumerate(cats)} return clsid2catid, catid2name
def widerfaceall_category_info(with_background=True): """ Get class id to category id map and category id to category name map of mixup wider_face dataset Args: with_background (bool, default True): whether load background as class 0. """ label_map = widerface_label(with_background) label_map = sorted(label_map.items(), key=lambda x: x[1]) cats = [l[0] for l in label_map] if with_background: cats.insert(0, 'background') clsid2catid = {i: i for i in range(len(cats))} catid2name = {i: name for i, name in enumerate(cats)} return clsid2catid, catid2name