概要
note
この文書は AI によって翻訳されています。内容に不正確な点や改善すべき点がございましたら、文書下部のコメント欄または以下の Issue ページにてご報告ください。
https://github.com/Seeed-Studio/wiki-documents/issues
概要
SSCMAは現在以下のモデルをサポートしています。対応するチュートリアルを参照してモデルのトレーニングを完了し、モデルの重みを取得することができます。
物体検出 - SWIFT-YOLO: デジタルメーターの読み取り
物体検出 - FOMO: マスク着用検出
キーポイント検出 PFLD: ポインターメーターの読み取り
パラメーターの説明
モデルトレーニング中の詳細なパラメーターについては、以下のコードを参照してください。
python3 tools/train.py --help
# SSCMAモデルのトレーニング
# 必須引数:
# config モデル設定ファイルのパス
# オプション引数:
# -h, --help このヘルプメッセージを表示して終了
# --work_dir WORK_DIR, --work-dir WORK_DIR
# ログとモデルを保存するディレクトリ
# --amp トレーニング中に自動混合精度を有効化 (https://pytorch.org/tutorials/recipes/recipes/amp_recipe.html)
# --auto_scale_lr, --auto-scale-lr
# トレーニング中に自動スケールLRを有効化
# --resume [RESUME] 最後のエポックのチェックポイント (または指定されたチェックポイントパス) からトレーニングを再開
# --no_validate, --no-validate
# トレーニング中のチェックポイント評価を無効化
# --launcher {none,pytorch,slurm,mpi}
# MMEngineのジョブランチャー
# --cfg_options CFG_OPTIONS [CFG_OPTIONS ...], --cfg-options CFG_OPTIONS [CFG_OPTIONS ...]
# 使用される設定の一部を上書き、'xxx=yyy'形式のキーと値のペアが設定ファイルに統合される
# --local_rank LOCAL_RANK, --local-rank LOCAL_RANK
# PyTorchのローカルランクを設定
# --dynamo_cache_size DYNAMO_CACHE_SIZE, --dynamo-cache-size DYNAMO_CACHE_SIZE
# PyTorchのダイナモキャッシュサイズ制限を設定
# --input_shape INPUT_SHAPE [INPUT_SHAPE ...], --input-shape INPUT_SHAPE [INPUT_SHAPE ...]
# 拡張: モデルパラメーター推定のための入力データ形状、例: 1 3 224 224
デプロイ
モデルをエクスポートした後、エッジコンピューティングデバイスにモデルをデプロイしてテストと評価を行うことができます。Deployセクションを参照して、モデルのデプロイ方法について詳しく学んでください。