Closed hushuai1 closed 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 , 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.
先按照ReadMe.md中的installation检察一下各个环境的版本是否匹配
感谢。确实是环境问题
请问是怎么解决的呢
您好,我在用您的demo进行测试的时候遇到了这个问题, 好像是这句代码的问题,求指点
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
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