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([
])
}