close
Skip to content

[Fix] make category_id in CocoWholeBodyDataset as numpy.array#2963

Merged
ly015 merged 1 commit intoopen-mmlab:dev-1.xfrom
zgjja:dev-1.x
Apr 23, 2024
Merged

[Fix] make category_id in CocoWholeBodyDataset as numpy.array#2963
ly015 merged 1 commit intoopen-mmlab:dev-1.xfrom
zgjja:dev-1.x

Conversation

@zgjja
Copy link
Copy Markdown
Contributor

@zgjja zgjja commented Feb 23, 2024

Motivation

parse_data_info() in class CocoWholeBodyDataset will return category_id as a List, making the transform() in class FilterAnnotations fails to indexing like a np.array.

Modification

make category_id as a numpy.array, just like what in class BaseCocoStyleDataset does.

Checklist

Before PR:

  • I have read and followed the workflow indicated in the CONTRIBUTING.md to create this PR.
  • Pre-commit or linting tools indicated in CONTRIBUTING.md are used to fix the potential lint issues.
  • Bug fixes are covered by unit tests, the case that causes the bug should be added in the unit tests.
  • New functionalities are covered by complete unit tests. If not, please add more unit tests to ensure correctness.
  • The documentation has been modified accordingly, including docstring or example tutorials.

After PR:

  • CLA has been signed and all committers have signed the CLA in this PR.

@CLAassistant
Copy link
Copy Markdown

CLAassistant commented Feb 23, 2024

CLA assistant check
All committers have signed the CLA.

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