YOLOv5バックボーンネットワークの実装
データセット準備
import torch
import torch.nn as nn
from torchvision import transforms, datasets
# デバイス設定
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
# データ変換処理
training_transforms = transforms.Compose([
transforms.Resize([224, 224]),
transforms.ToTensor(),
transforms.Normalize(mean=[0.485, 0 ...
7月10日 18:14 投稿