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

Vitis AI ユーザー ガイド (UG1414)

Document ID
UG1414
Release Date
2023-09-28
Version
3.5 日本語

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

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