Skip to content

ML.NET ObjectDetection vs YOLOv4 #7230

Open

Description

Is your feature request related to a problem? Please describe.
Why is ML.NET significantly worse than YOLOv4(AlexeyAB) in accuracy?
Is the ability of the AutoFormer algorithm used by ML.NET to detect objects inherently weaker than YOLOv4(AlexeyAB)?
In the ML.NET training parameter Epoch, the accuracy will not be better with more times. Is there any way to improve the accuracy of ML.NET?

Describe the solution you'd like
My training dataset has 16 images and my test dataset has 4 images.
The purpose is to identify specific mark in grayscale images, such as the cross mark in the picture below.
Comparing object detection ML.NET(Version: 16.18 for VS2019) and YOLOv4(AlexeyAB).

Describe alternatives you've considered
YOLOv4(AlexeyAB) can detect the mark correctly in all 4 images of the test dataset.
The ML.NET training parameters use the default value Epoch 5. Only 2 images can be detected correctly.
If the Epoch is changed to 50 or 500, 4 images cannot be detected.

Additional context
mark_sample

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or requestuntriagedNew issue has not been triaged

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions