Modelos TensorFlow
Os seguintes modelos pré-treinados estão disponíveis para uso no aprendizado por transferência com o algoritmo de detecção de objetos - TensorFlow:
Os modelos a seguir variam significativamente em tamanho, número de parâmetros do modelo, tempo de treinamento e latência de inferência para qualquer conjunto de dados. O melhor modelo para seu caso de uso depende da complexidade do seu conjunto de dados de ajuste fino e de quaisquer requisitos que você tenha sobre tempo de treinamento, latência de inferência ou precisão do modelo.
| Nome do modelo | model_id |
Origem |
|---|---|---|
ResNet50 V1 VPN 640 |
|
|
EfficientDet D0 512 |
|
|
EfficientDet D1 640 |
|
|
EfficientDet D2 768 |
|
|
EfficientDet D3 896 |
|
|
MobileNet V1 VPN 640 |
|
|
MobileNet V2 FPNLite 320 |
|
|
MobileNet V2 FPNLite 640 |
|
|
ResNet50 V1 VPN 1024 |
|
|
ResNet101 V1 FPN 640 |
|
|
ResNet101 V1 FPN 1024 |
|
|
ResNet152 V1 FPN 640 |
|
|
ResNet152 V1 FPN 1024 |
|
