WindowsでAIプログラミングと動画作成

Pythonライブラリ:Pytorch(作成中)

torch

GPU・CUDA関連:TORCH.CUDA

PyTorchでGPUが使用可能か確認

torch.cuda.is_available()

PyTorchで使用できるGPU(デバイス)数の確認

torch.cuda.device_count()

GPUの名称

torch.cuda.get_device_name()

CUDA Compute Capability

torch.cuda.get_device_capability()

機械学習

データセット

torch.utils.data.Dataset

データローダー

TORCH.UTILS.DATA

torch.utils.data.DataLoader

torch.nn

torch.nn.Module

torchvision

コマンド内容入力出力
Compose複数の Transform を連続して行うPIL ImagePIL Image
RandomApply複数の Transform を指定した確率で行うPIL ImagePIL Image
RandomChoice複数の Transform から1つを選択して行うPIL ImagePIL Image
RandomOrder複数の Transform をランダムに順番を入れ替えて行うPIL ImagePIL Image
Padパディングを行うPIL ImagePIL Image
CenterCrop画像の中心を切り抜くPIL ImagePIL Image
FiveCrop4隅及び中心の計5箇所を切り抜くPIL ImagePIL Image のタプル
TenCrop元の画像及び左右反転した画像の4隅及び中心の計10箇所を切り抜くPIL ImagePIL Image のタプル
Resizeリサイズを行うPIL ImagePIL Image
Grayscaleグレースケール変換を行うPIL ImagePIL Image
RandomCropランダムに画像を切り抜くPIL ImagePIL Image
RandomResizedCropランダムにリサイズ及び切り抜きを行うPIL ImagePIL Image
ColorJitterランダムに明るさ、コントラスト、彩度、色相を変化させるPIL ImagePIL Image
RandomGrayscaleランダムにグレースケール変換を行うPIL ImagePIL Image
RandomHorizontalFlipランダムに左右反転を行うPIL ImagePIL Image
RandomVerticalFlipランダムに上下反転を行うPIL ImagePIL Image
RandomAffineランダムにアフィン変換を行うPIL ImagePIL Image
RandomPerspectiveランダムに射影変換を行うPIL ImagePIL Image
RandomRotationランダムに回転を行うPIL ImagePIL Image
LinearTransformation線形変換を行うテンソルテンソル
Normalize標準化を行うテンソルテンソル
RandomErasingランダムに選択した領域を除去するテンソルテンソル
ToPILImageテンソルをPIL Image オブジェクトに変換するテンソルPIL Image
ToTensorPIL Image オブジェクトをテンソルに変換するPIL Imageテンソル
Lambdaユーザー定義の Transform を作成する場合に利用する任意のオブジェクト任意のオブジェクト