def __init__(self): self.data_transform = { 'train': Compose([ get_anno(), # JSONからアノテーションを辞書に格納 add_neck(), # アノテーションデータの順番を変更し、さらに首のアノテーションデータを追加 aug_scale(), # 拡大縮小 aug_rotate(), # 回転 aug_croppad(), # 切り出し aug_flip(), # 左右反転 remove_illegal_joint(), # 画像からはみ出たアノテーションを除去 Normalize_Tensor() # 色情報の標準化とテンソル化 ]), 'val': Compose([ get_anno(), # JSONからアノテーションを辞書に格納 add_neck(), # アノテーションデータの順番を変更し、さらに首のアノテーションデータを追加 aug_scale(), # 拡大縮小 aug_rotate(), # 回転 aug_croppad(), # 切り出し aug_flip(), # 左右反転 remove_illegal_joint(), # 画像からはみ出たアノテーションを除去 Normalize_Tensor() # 色情報の標準化とテンソル化 ]) }
def __init__(self): self.data_transform = { 'train': Compose([ get_anno(), add_neck(), aug_scale(), aug_rotate(), aug_croppad(), aug_flip(), remove_illegal_joint(), Normalize_Tensor() ]), 'val': Compose([ ]) }