CFGでTTA定義 datasetでもself.TTA定義 train roopの際に画像をそれぞれ受け取ってGPUに乗せてmodelに入れて平均をとる class TestData(Dataset): def __init__(self, TTA=False): self.TTA = TTA def __getitem__(self,index): if self.TTA: imgs = ... imgs2 = imgs.flip(2,3) # 縦横反転(2が縦、3が横) imgs3 = torch.rot90(imgs,1,[2,3]) #2,3次元の軸で左に90度回転 imgs4 = t…