PyTorch実戦入門 第2章 演習問題 解答まとめ
1
a
必要な前処理は,第2章で行ったことと同じで,256×256にリサイズする事と,テンソルに変換する事です.
preprocess = transforms.Compose([transforms.Resize(256), transforms.ToTensor()]) img = Image.open("golden.png") img_t = preprocess(img)
b
入力したゴールデンレトリーバーの画像は以下になります.
出力結果は以下になります.
上のゴールデンレトリーバーはウマと認識されているのかシマ模様が出来ています.下のゴールデンレトリーバーはウマの学習データに,このような寝そべった体勢の画像が無かったからか耳など一部しかシマ模様になっていないことが分かります.
2
b
例えばこのリポジトリのプロジェクトがあります.
github.com
yoloとは物体検出を行うモデルであり,yolov5その最新のモデルになります.