モデルを作成する - 2.5 日本語

Vitis AI オプティマイザー ユーザー ガイド (UG1333)

Document ID
UG1333
Release Date
2022-06-15
Version
2.5 日本語

簡単にするために、ここでは torchvision の mobilenet_v2 を使用します。

from torchvision.models.mobilenet import mobilenet_v2
model = mobilenet_v2(pretrained=True)