YOLOv5 Backboneモジュールの実装
この記事では、YOLOv5のBackboneモジュールを実装する手順を説明します。
環境設定
言語: Python 3.8
開発環境: PyCharm
データセット: 天気予測データセット（参照: 深度学習Day-03）
ライブラリ: torch==1.12.1+cu113, torchvision==0.13.1+cu113
初期設定
1. GPUの設定
import torch
import torch.nn as nn
from torchvision import transforms, datasets
import warn ...
8月28日 14:46 投稿