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

MaskDINO Refactor #4006

Merged
merged 119 commits into from
Oct 28, 2024
Merged

Conversation

eugene123tw
Copy link
Collaborator

@eugene123tw eugene123tw commented Oct 8, 2024

Summary

MaskDINO refactor PR specifically include the following:

  • Replaced detectron2.resnet with tv.resnet50
  • Organized shared files (e.g., position embeddings, visual encoder/decoder) under a common folder
  • Refactored the Hungarian matcher to incorporate mask cost and dice cost
  • Enhanced MaskDINOHead class with export and post-process functionality
  • Removed unnecessary code (batch norm, utility functions, box_ops, etc.) after refactoring
  • Modularized various MaskDINO head classes (maskdino_decoder, maskdino_encoder, maskdino_head)
  • Improved unit test cases for the MaskDINOHead, encoder, and decoder components

How to test

Checklist

  • I have added unit tests to cover my changes.​
  • I have added integration tests to cover my changes.​
  • I have ran e2e tests and there is no issues.
  • I have added the description of my changes into CHANGELOG in my target branch (e.g., CHANGELOG in develop).​
  • I have updated the documentation in my target branch accordingly (e.g., documentation in develop).
  • I have linked related issues.

License

  • I submit my code changes under the same Apache License that covers the project.
    Feel free to contact the maintainers if that's a concern.
  • I have updated the license header for each file (see an example below).
# Copyright (C) 2024 Intel Corporation
# SPDX-License-Identifier: Apache-2.0

@eugene123tw eugene123tw requested review from sungchul2, kprokofi and eunwoosh and removed request for sungchul2 and kprokofi October 22, 2024 15:31
@sungchul2
Copy link
Contributor

sungchul2 commented Oct 24, 2024

@eugene123tw could you describe the purpose and changes of this PR in Summary, and update CHANGELOG?

@github-actions github-actions bot added the DOC Improvements or additions to documentation label Oct 24, 2024
sungchul2
sungchul2 previously approved these changes Oct 25, 2024
@eugene123tw eugene123tw enabled auto-merge (squash) October 25, 2024 20:14
@eugene123tw eugene123tw merged commit f08e891 into openvinotoolkit:develop Oct 28, 2024
20 of 21 checks passed
@eugene123tw eugene123tw deleted the eugene/maskdino-optim branch October 28, 2024 08:35
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
DOC Improvements or additions to documentation OTX 2.0 TEST Any changes in tests
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants