Skip to content

Update DeepLabV3 input images to 520x520 #830

@msluszniak

Description

@msluszniak

Currently our model is exported to accept images of size 224x224. This is incorrect since DeepLabV3 was trained to work on images of size 520x520. We need to swap model on HuggingFace to provide the correct one. For example, the segmentation does not work correctly on the following image:

Image

Metadata

Metadata

Assignees

Labels

modelIssues related to exporting, improving, fixing ML models

Type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions