Replace ailia.get_gpu_environment_id() to args.env_id

axinc-ai / ailia-models

The collection of pre-trained, state-of-the-art AI models for ailia SDK

2.07k stars 330 forks source link

Replace ailia.get_gpu_environment_id() to args.env_id #601

Closed kyakuno closed 3 years ago

kyakuno commented 3 years ago

直近で追加いただいていくつかのモデルがailia.get_gpu_environment_id()を使用してenv_idを取得しているようです。現在は、utils.pyでargs.env_idにコマンドラインで与えた値が入るようになっていますので、こちらに置き換えをお願いいたします。

kyakuno commented 3 years ago

置き換え例 https://github.com/axinc-ai/ailia-models/pull/600/commits/921a941d322b28f2316ea2f9e3ebf73bce0901cc

kyakuno commented 3 years ago

対象モデル auto_speech transformer-cnn-emotion-recognition u2net-human-seg age-gender-recognition-retail minimal-hand wide_resnet50 3d-object-detection.pytorch mediapipe_objectron animalpose blazepose-fullbody road-segmentation-adas roneld vehicle-attributes-recognition-barrier vehiecle-licence-plate-detection-barrier