search

HZAI-ZJNU / Mamba-YOLO

the official pytorch implementation of “Mamba-YOLO：SSMs-based for Object Detection”
Apache License 2.0
227 stars 27 forks source link

Can this model perform classification tasks? Is there any YAML file for reference? #35

Open Sine7812 opened 2 months ago

Sine7812 commented 2 months ago

Can this model perform classification tasks? Is there any YAML file for reference?

EthanW-coder commented 2 months ago

Mamba-YOLO is based on the ultralytics implementation, we have not done any experiments on this, you can follow the classification yaml provided by ultralytics to modify, but this method is inefficient. It is worth mentioning that we are in the process of implementing a version of Mamba-YOLO based on mmdet, and may do experiments on it at that time.