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

DDRNet for Semantic Segmentation #70

Merged
merged 114 commits into from
Sep 20, 2024
Merged

DDRNet for Semantic Segmentation #70

merged 114 commits into from
Sep 20, 2024

Conversation

sokovninn
Copy link
Member

This PR integrates the DDRNet model to enhance semantic segmentation performance.

Comparison with Current Model (MicroNet + Segmentation Head)

Parameters: DDRNet has 5.7M, compared to 2.1M for MicroNet.
Training Time: Both models have similar single-epoch training times.

Metrics After 1 Epoch

MicroNet + Segmentation Head:

  • Jaccard Index: 0.00873
  • F1 Score: 0.69811

DDRNet-23-slim:

  • Jaccard Index: 0.00952
  • F1 Score: 0.70069

Possible improvements

  • Compute Jaccard Index every n epochs to increase training speed.
  • Improve handling of auxiliary heads for export.
  • Use pretrained weights.
  • Add Online Hard Example Mining (OHEM) with Cross Entropy Loss.
  • Add EMA (Exponential Moving Average).

Base automatically changed from cleanup to dev September 19, 2024 05:06
@kozlov721 kozlov721 changed the title Add DDRNet model for semantic segmentation DDRNet for Semantic Segmentation Sep 19, 2024
Copy link

github-actions bot commented Sep 19, 2024

Test Results

  1 files    1 suites   21m 36s ⏱️
129 tests 129 ✅ 0 💤 0 ❌
180 runs  180 ✅ 0 💤 0 ❌

Results for commit 438c7a3.

♻️ This comment has been updated with latest results.

Copy link
Collaborator

@kozlov721 kozlov721 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

Copy link
Contributor

@klemen1999 klemen1999 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@sokovninn sokovninn merged commit 2449850 into dev Sep 20, 2024
@sokovninn sokovninn deleted the feature/segmentation-ddrnet branch September 20, 2024 11:35
kozlov721 added a commit that referenced this pull request Oct 9, 2024
Co-authored-by: Martin Kozlovsky <martin.kozlovsky@luxonis.com>
Co-authored-by: GitHub Actions <actions@github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request tests Adding or changing tests
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants