THU-MIG / yolov10

YOLOv10: Real-Time End-to-End Object Detection [NeurIPS 2024]
https://arxiv.org/abs/2405.14458
GNU Affero General Public License v3.0
9.58k stars 913 forks source link

Error while importing YOLOv10 on kaggle #285

Closed hmasum52 closed 3 months ago

hmasum52 commented 3 months ago
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
Cell In[32], line 2
      1 get_ipython().system(' cd {HOME}')
----> 2 from ultralytics import YOLOv10
      4 model = YOLOv10(f'{HOME}/weights/yolov10n.pt')
      5 model.info()

File /kaggle/working/yolov10/ultralytics/__init__.py:6
      3 __version__ = "8.1.34"
      5 from ultralytics.data.explorer.explorer import Explorer
----> 6 from ultralytics.models import RTDETR, SAM, YOLO, YOLOWorld, YOLOv10
      7 from ultralytics.models.fastsam import FastSAM
      8 from ultralytics.models.nas import NAS

File /kaggle/working/yolov10/ultralytics/models/__init__.py:6
      4 from .sam import SAM
      5 from .yolo import YOLO, YOLOWorld
----> 6 from .yolov10 import YOLOv10
      8 __all__ = "YOLO", "RTDETR", "SAM", "YOLOWorld", "YOLOv10"  # allow simpler import

File /kaggle/working/yolov10/ultralytics/models/yolov10/__init__.py:1
----> 1 from .model import YOLOv10
      2 from .predict import YOLOv10DetectionPredictor
      3 from .val import YOLOv10DetectionValidator

File /kaggle/working/yolov10/ultralytics/models/yolov10/model.py:10
      7 from huggingface_hub import PyTorchModelHubMixin
      8 from .card import card_template_text
---> 10 class YOLOv10(Model, PyTorchModelHubMixin, model_card_template=card_template_text):
     12     def __init__(self, model="yolov10n.pt", task=None, verbose=False, 
     13                  names=None):
     14         super().__init__(model=model, task=task, verbose=verbose)

File /opt/conda/lib/python3.10/site-packages/huggingface_hub/hub_mixin.py:619, in PyTorchModelHubMixin.__init_subclass__(cls, tags, *args, **kwargs)
    617 tags.append("pytorch_model_hub_mixin")
    618 kwargs["tags"] = tags
--> 619 return super().__init_subclass__(*args, **kwargs)

TypeError: ModelHubMixin.__init_subclass__() got an unexpected keyword argument 'model_card_template'

I am getting this traceback while running the following code in kaggle

from ultralytics import YOLOv10

model = YOLOv10(f'{HOME}/weights/yolov10n.pt')

faced the same issue with yolo command also image

hmasum52 commented 3 months ago

It was a Kaggle environment issue. The problem resolved after setting the environment to always use latest version image