2019年3月12日火曜日

Coral USB Accelerator(Edge TPU)の学習済みモデル

Edge TPUで用意されているImage classificationとObject detectionモデルを整理する。
ベータ版で公開されているモデルはここを参照
All modelsをダウンロードし、解凍するとtxtファイルとtfliteファイルが入っている。

  • txtファイル
    ラベルの番号と名称が対応した2スペース区切り?のファイルである。
    学習済みモデルにはないがOxford-IIIT Pet Datasetのラベル(pet_labels.txt)もある。
    これは、fine-tuningのチュートリアルで使うのだろう?
  • tfliteファイル
    学習済みモデル。
    *_quant_edgetpu.tflite → Edge TPU向けの学習済みモデル
    *_quant.tflite → TF-Lite向けの学習済みモデル。CPUで動作できる。

Image classification

  • MobileNet V1 (ImageNet)
    mobilenet_v1_1.0_224_quant.tflite
    mobilenet_v1_1.0_224_quant_edgetpu.tflite
  • MobileNet V2 (ImageNet)
    mobilenet_v2_1.0_224_quant.tflite
    mobilenet_v2_1.0_224_quant_edgetpu.tflite
  • MobileNet V2 (iNat insects) → iNaturalist datasetの昆虫
    mobilenet_v2_1.0_224_inat_insect_quant.tflite
    mobilenet_v2_1.0_224_inat_insect_quant_edgetpu.tflite
  • MobileNet V2 (iNat plants) → iNaturalist datasetの植物
    mobilenet_v2_1.0_224_inat_plant_quant.tflite
    mobilenet_v2_1.0_224_inat_plant_quant_edgetpu.tflite
  • MobileNet V2 (iNat birds) → iNaturalist datasetの鳥
    mobilenet_v2_1.0_224_inat_bird_quant.tflite
    mobilenet_v2_1.0_224_inat_bird_quant_edgetpu.tflite
  • Inception V1 (ImageNet)
    inception_v1_224_quant.tflite
    inception_v1_224_quant_edgetpu.tflite
  • Inception V2 (ImageNet)
    inception_v2_224_quant.tflite
    inception_v2_224_quant_edgetpu.tflite
  • Inception V3 (ImageNet)
    inception_v3_299_quant.tflite
    inception_v3_299_quant_edgetpu.tflite
  • Inception V4 (ImageNet)
    inception_v4_299_quant.tflite
    inception_v4_299_quant_edgetpu.tflite

Object detection

  • MobileNet SSD v1 (COCO)
    mobilenet_ssd_v1_coco_quant_postprocess.tflite
    mobilenet_ssd_v1_coco_quant_postprocess_edgetpu.tflite
  • MobileNet SSD v2 (COCO)
    mobilenet_ssd_v2_coco_quant_postprocess.tflite
    mobilenet_ssd_v2_coco_quant_postprocess_edgetpu.tflite
  • MobileNet SSD v2 (Faces)
    mobilenet_ssd_v2_face_quant_postprocess.tflite
    mobilenet_ssd_v2_face_quant_postprocess_edgetpu.tflite
(Object detectionの学習済みモデルの"postprocess"はなんの意味があるのだろうか?)

Embedding extractor (classification)

CPU側で学習できるモデルのようだ・・・
  • MobileNet v1
    mobilenet_v1_1.0_224_quant_embedding_extractor_edgetpu.tflite

0 件のコメント:

コメントを投稿