Add the ORTModelForSemanticSegmentation class #539
JingyaHuang
merged
15 commits into
huggingface:main
from
TheoMrc:ORTModelForImageSegmentation-implementation
Initial commit for ORTModelForImageSegmentation
27e62e6d
Example name and io binding output size fix
b2e6fc6f
TheoMrc
changed the title
Refactor to ORTModelForSemanticSegmentation
e59fa8aa
Merge branch 'main' into ORTModelForImageSegmentation-implementation
36b4c6b7
Merge remote-tracking branch 'upstream/main' into ORTModelForImageSeg…
1dcf8c1a
IOBindingHelper for io_binding in ORTModelForSemanticSegmentation
ad6300dc
Black and isort formatting
ff63f5a3
Apply various suggestions from code review
29ef614c
Alphanumeric order in modeling_ort init
4bc1befe
Fixing docstring model name and removing comments
47e85462
Fixing tests and add import
f7dc114e
Black formatting and fixed unmatched quote
e21e721b
Removing export_feature class attribute
f144ae3f
Fixing test model name
f765bfc4
Adding 'image-segmentation' task to optimum/pipelines and fixing tests
c6420dae
JingyaHuang
approved these changes
on 2022-12-15
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub