hukaixuan19970627 / yolov5_obb

yolov5 + csl_label.(Oriented Object Detection)(Rotation Detection)(Rotated BBox)基于yolov5的旋转目标检测
GNU General Public License v3.0
1.83k stars 427 forks source link

RuntimeError: Could not infer dtype of numpy.float32 #60

Closed hushuai1 closed 3 years ago

hushuai1 commented 3 years ago

您好,我在用您的demo进行测试的时候遇到了这个问题, 好像是这句代码的问题,求指点 c = torch.tensor(labels[:, 0]) # classes Scanning labels DOTA_demo_view/labels.cache (44 found, 0 missing, 0 empty, 0 duplicate, for 44 images): 44it [00:00, 13217.03it/s] Traceback (most recent call last): File "train.py", line 724, in train(hyp, opt, device, tb_writer) File "train.py", line 311, in train c = torch.tensor(labels[:, 0]) # classes RuntimeError: Could not infer dtype of numpy.float32 image

github-actions[bot] commented 3 years ago

Hello @hushuai1, thank you for your interest in our work! Please visit our Custom Training Tutorial to get started, and see our Jupyter Notebook Open In Colab, Docker Image, and Google Cloud Quickstart Guide for example environments.

If this is a bug report, please provide screenshots and minimum viable code to reproduce your issue, otherwise we can not help you.

If this is a custom model or data training question, please note Ultralytics does not provide free personal support. As a leader in vision ML and AI, we do offer professional consulting, from simple expert advice up to delivery of fully customized, end-to-end production solutions for our clients, such as:

For more information please visit https://www.ultralytics.com.

hukaixuan19970627 commented 3 years ago

先按照ReadMe.md中的installation检察一下各个环境的版本是否匹配

hushuai1 commented 3 years ago

感谢。确实是环境问题

baimaofulvshui commented 2 months ago

请问是怎么解决的呢