ベータ版で公開されているモデルはここを参照。
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 件のコメント:
コメントを投稿