INNER CODE UNIT · Python

format_dict

zhulinyv/Semi-Auto-NovelAI-to-Pixiv · src/batch_tagger.py:33

def format_dict(dict_):
    try:
        list_ = dict_["confidences"]
        dict_ = {}
        for i in list_:
            dict_.update({i["label"]: i["confidence"]})
        return dict_
    except KeyError:
        return None


def tagger(
    image,
    path,
    batch,
    model_repo,
    general_thresh,
    general_mcut_enabled,

View source record →