探索 Model Garden

Model Garden 提供了一系列先进的机器学习 (ML) 模型，可用于实现视觉、文本和音频功能。这些模型经过优化，可与 Google Tensor SDK 搭配使用，让您能够直接在 Pixel 设备上实现强大的 AI 功能，并获得流畅的设备端性能。

深度估测

模型 许可
depth_anything_v2 Apache-2.0
midas_v2_1 BSD-3-Clause

面部重建

模型 许可
facemap_3dmm BSD-3-Clause

图片和文本理解

模型 许可
clip 麻省理工学院
mobileclip_image_encoder 麻省理工学院
mobileclip_text_encoder 麻省理工学院
tinyclip 麻省理工学院

图片分类

模型 许可
alexnet BSD-3-Clause
beit BSD-3-Clause
convnext_base BSD-3-Clause
convnext_tiny BSD-3-Clause
densenet121 BSD-3-Clause
efficientformer_l1 Apache-2.0
efficientformerv2_s0 Apache-2.0
efficientnet_b0 BSD-3-Clause
efficientnet_b1 BSD-3-Clause
efficientnet_b2 BSD-3-Clause
efficientnet_b3 BSD-3-Clause
efficientnet_b4 BSD-3-Clause
efficientnet_b5 BSD-3-Clause
efficientnet_b6 BSD-3-Clause
efficientnet_b7 BSD-3-Clause
efficientnet_v2_s BSD-3-Clause
efficientnetv2_m APACHE-2.0
efficientvit_cls_b2 BSD-3-Clause
efficientvit_cls_l2 BSD-3-Clause
efficientvit_seg_l2 APACHE-2.0
googlenet BSD-3-Clause
inception_v3 BSD-3-Clause
levit APACHE-2.0
maxvit_t BSD-3-Clause
mnasnet0 BSD-3-Clause
mobile_vit BSD-3-Clause
mobilenet_v2 APACHE-2.0
mobilenet_v3_large BSD-3-Clause
mobilenet_v3_small BSD-3-Clause
mobilenetv4_conv_l BSD-3-Clause
mobilenetv4_conv_m BSD-3-Clause
mobilenetv4_conv_s BSD-3-Clause
mobilenetv4_hybrid_l BSD-3-Clause
mobilenetv4_hybrid_medium APACHE-2.0
nfnet BSD-3-Clause
pvt_v2_b1 BSD-3-Clause
pvt_v2_b3 BSD-3-Clause
regnety APACHE-2.0
resnest14d BSD-3-Clause
resnet101 BSD-3-Clause
resnet152 BSD-3-Clause
resnet18 BSD-3-Clause
resnet50 BSD-3-Clause
resnext101 AI-HUB-MODELS
resnext50 BSD-3-Clause
shufflenet_v2 BSD-3-Clause
squeezenet1 BSD-3-Clause
swin_small BSD-3-Clause
swin_tiny BSD-3-Clause
tf_efficientnetv2_m APACHE-2.0
vgg16 BSD-3-Clause
vit_base_patch16 APACHE-2.0
vit_small_patch16 BSD-3-Clause
wide_resnet101 BSD-3-Clause
wide_resnet50 BSD-3-Clause

图像分割

模型 许可
hrnet_w48_ocr 麻省理工学院
mediapipe_selfie APACHE-2.0
unet_segmentation GPL-3.0

图像超分辨率

模型 许可
esrgan APACHE-2.0

对象检测

模型 许可
3d_deep_box 麻省理工学院
conditional_detr_resnet50 Apache-2.0
detr_resnet50 Apache-2.0
detr_resnet50_dc5 Apache-2.0
detr_resnet101 Apache-2.0
detr_resnet101_dc5 Apache-2.0
faceattribnet AI-HUB-MODELS
lightweight_face_detection AI-HUB-MODELS
mediapipe_hand_detection APACHE-2.0
person_foot_detection AI-HUB-MODELS
ppe_detection AI-HUB-MODELS
yolo_v4 Apache-2.0
yolo_v6 GPL-3.0
yolo_v7 GPL-3.0
yolos_tiny APACHE-2.0
yolox_tiny APACHE-2.0

姿势估计

模型 许可
hrnet_pose 麻省理工学院
lite_hrnet_pose APACHE-2.0
mediapipe_pose APACHE-2.0
movenet 麻省理工学院

问答

模型 许可
tinyroberta CC-BY-4.0

语义分割

模型 许可
bgnet Apache-2.0
bisenet 无许可文件
ddrnet23_slim 麻省理工学院
deeplabv3_mobilenet_v3_large BSD-3-Clause
deeplabv3_plus_mobilenet 麻省理工学院
deeplabv3_resnet101 BSD-3-Clause
deeplabv3_resnet50 BSD-3-Clause
fcn_resnet50 BSD-3-Clause
ffnet_122ns_lowres BSD-3-Clause
ffnet_40s BSD-3-Clause
ffnet_54s BSD-3-Clause
ffnet_78s_lowres BSD-3-Clause
isnet Apache 2.0
lraspp_mobilenet_v3_large BSD-3-Clause
sam_vit_b APACHE-2.0
sam_vit_l APACHE-2.0
segformer NVIDIA-SCSL
segment_anything_model Apache-2.0
u2net_full APACHE-2.0
u2net_lite APACHE-2.0

语音识别

模型 许可
deepspeech BSD-2-Clause
torchaudio_emformer_rnnt_base BSD-2-Clause
wav2vec2_base_960h APACHE-2.0

超分辨率

模型 许可
quicksrnet_large BSD-3-Clause
quicksrnet_small BSD-3-Clause
real_esrgan_general_x4v3 BSD-3-Clause
real_esrgan_x4plus BSD-3-Clause
xlsr BSD-3-Clause

文本分类

模型 许可
distilbert Apache-2.0
mobilebert Apache-2.0