Augmix در Pytorch (1) – Community Dev

from torchvision.datasets import OxfordIIITPet
from torchvision.transforms.v2 import AugMix
from torchvision.transforms.functional import InterpolationMode

am = AugMix()
am = AugMix(severity=3, mixture_width=3, chain_depth=-1, alpha=1.0, 
            all_ops=True, interpolation=InterpolationMode.BILINEAR,
            fill=None)
am
# AugMix(interpolation=InterpolationMode.BILINEAR, severity=3,
#        mixture_width=3, chain_depth=-1, alpha=1.0, all_ops=True)

am.severity
# 3

am.mixture_width
# 3

am.chain_depth
# -1

am.alpha
# 1.0

am.all_ops
# True

am.interpolation
# 

print(am.fill)
# None

origin_data = OxfordIIITPet(
    root="data",
    transform=None
)

default_data = OxfordIIITPet(
    root="data",
    transform=AugMix()
)

aoFalse_data = OxfordIIITPet( # `ao` is all_ops.
    root="data",
    transform=AugMix(all_ops=False)
    # transform=AugMix(severity=3, mixture_width=3, chain_depth=-1, 
    #                  alpha=1.0, all_ops=True,
    #                  interpolation=InterpolationMode.BILINEAR,
    #                  fill=None)
)

s10cd25fgray_data = OxfordIIITPet( # `s` is severity and `cd` is chain_depth.
    root="data",                   # `f` is fill.
    transform=AugMix(severity=10, chain_depth=25, fill=150)
)

s10cd25fpurple_data = OxfordIIITPet(
    root="data",
    transform=AugMix(severity=10, chain_depth=25, fill=[160, 32, 240])
)

import matplotlib.pyplot as plt

def show_images1(data, main_title=None):
    plt.figure(figsize=[10, 5])
    plt.suptitle(t=main_title, y=0.8, fontsize=14)
    for i, (im, _) in zip(range(1, 6), data):
        plt.subplot(1, 5, i)
        plt.imshow(X=im)
        plt.xticks(ticks=[])
        plt.yticks(ticks=[])
    plt.tight_layout()
    plt.show()

show_images1(data=origin_data, main_title="origin_data")
print()
show_images1(data=default_data, main_title="default_data")
show_images1(data=default_data, main_title="default_data")
show_images1(data=default_data, main_title="default_data")
show_images1(data=default_data, main_title="default_data")
show_images1(data=default_data, main_title="default_data")
show_images1(data=default_data, main_title="default_data")
show_images1(data=default_data, main_title="default_data")
show_images1(data=default_data, main_title="default_data")
show_images1(data=default_data, main_title="default_data")
show_images1(data=default_data, main_title="default_data")
print()
show_images1(data=aoFalse_data, main_title="aoFalse_data")
show_images1(data=aoFalse_data, main_title="aoFalse_data")
show_images1(data=aoFalse_data, main_title="aoFalse_data")
show_images1(data=aoFalse_data, main_title="aoFalse_data")
show_images1(data=aoFalse_data, main_title="aoFalse_data")
show_images1(data=aoFalse_data, main_title="aoFalse_data")
show_images1(data=aoFalse_data, main_title="aoFalse_data")
show_images1(data=aoFalse_data, main_title="aoFalse_data")
show_images1(data=aoFalse_data, main_title="aoFalse_data")
show_images1(data=aoFalse_data, main_title="aoFalse_data")
print()
show_images1(data=s10cd25fgray_data, main_title="s10cd25fgray_data")
show_images1(data=s10cd25fpurple_data, main_title="s10cd25fpurple_data")

# ↓ ↓ ↓ ↓ ↓ ↓ The code below is identical to the code above. ↓ ↓ ↓ ↓ ↓ ↓
def show_images2(data, main_title=None, s=3, mw=3, cd=-1, a=1.0,
                 ao=True, ip=InterpolationMode.BILINEAR, f=None):
    plt.figure(figsize=[10, 5])
    plt.suptitle(t=main_title, y=0.8, fontsize=14)
    if main_title != "origin_data":
        for i, (im, _) in zip(range(1, 6), data):
            plt.subplot(1, 5, i)
            am = AugMix(severity=s, mixture_width=mw, chain_depth=cd,
                        alpha=a, all_ops=ao, interpolation=ip, fill=f)
            plt.imshow(X=am(im))
            plt.xticks(ticks=[])
            plt.yticks(ticks=[])
    else:
        for i, (im, _) in zip(range(1, 6), data):
            plt.subplot(1, 5, i)
            plt.imshow(X=im)
            plt.xticks(ticks=[])
            plt.yticks(ticks=[])
    plt.tight_layout()
    plt.show()

show_images2(data=origin_data, main_title="origin_data")
print()
show_images2(data=origin_data, main_title="default_data")
show_images2(data=origin_data, main_title="default_data")
show_images2(data=origin_data, main_title="default_data")
show_images2(data=origin_data, main_title="default_data")
show_images2(data=origin_data, main_title="default_data")
show_images2(data=origin_data, main_title="default_data")
show_images2(data=origin_data, main_title="default_data")
show_images2(data=origin_data, main_title="default_data")
show_images2(data=origin_data, main_title="default_data")
show_images2(data=origin_data, main_title="default_data")
print()
show_images2(data=origin_data, main_title="aoFalse_data", ao=False)
show_images2(data=origin_data, main_title="aoFalse_data", ao=False)
show_images2(data=origin_data, main_title="aoFalse_data", ao=False)
show_images2(data=origin_data, main_title="aoFalse_data", ao=False)
show_images2(data=origin_data, main_title="aoFalse_data", ao=False)
show_images2(data=origin_data, main_title="aoFalse_data", ao=False)
show_images2(data=origin_data, main_title="aoFalse_data", ao=False)
show_images2(data=origin_data, main_title="aoFalse_data", ao=False)
show_images2(data=origin_data, main_title="aoFalse_data", ao=False)
show_images2(data=origin_data, main_title="aoFalse_data", ao=False)
print()
show_images2(data=origin_data, main_title="s10cd25fgray_data", s=10, cd=25,
             f=150)
show_images2(data=origin_data, main_title="s10cd25fpurple_data", s=10, 
             cd=25, f=[160, 32, 240])