Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[CodeCamp #2] Add jax support to MeanIoU #67

Merged
merged 11 commits into from
Dec 19, 2022
Merged

[CodeCamp #2] Add jax support to MeanIoU #67

merged 11 commits into from
Dec 19, 2022

Conversation

xuan07472
Copy link
Contributor

@xuan07472 xuan07472 commented Dec 16, 2022

Motivation

Add jax support to MeanIoU

Modification

Calculated using jax.numpy.ndarray for MeanIoU support, supplemented and unit tested

@CLAassistant
Copy link

CLAassistant commented Dec 16, 2022

CLA assistant check
All committers have signed the CLA.

@ice-tong
Copy link
Collaborator

Please update the docstring of MeanIoU~

This metric supports 5 kinds of inputs, i.e. ``numpy.ndarray``,
``torch.Tensor``, ``oneflow.Tensor``, ``tensorflow.Tensor`` and
``paddle.Tensor``, and the implementation for the calculation depends

@ice-tong ice-tong changed the title CodeCamp #2 [CodeCamp #2] Add jax support to MeanIoU Dec 17, 2022
Co-authored-by: yancong <32220263+ice-tong@users.noreply.github.com>
Co-authored-by: yancong <32220263+ice-tong@users.noreply.github.com>
@ice-tong
Copy link
Collaborator

LGTM, and please @zhouzaida have a look~

限制最新发行的numpy1.24.0,保证oneflow的兼容性
补充修改,限制numpy版本,保证oneflow兼容性
@zhouzaida zhouzaida merged commit 466bcb6 into open-mmlab:main Dec 19, 2022
@xuan07472 xuan07472 deleted the branchname branch December 19, 2022 06:25
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants