search

opendatalab / DocLayout-YOLO

DocLayout-YOLO: Enhancing Document Layout Analysis through Diverse Synthetic Data and Global-to-Local Adaptive Perception
https://huggingface.co/spaces/opendatalab/DocLayout-YOLO
GNU Affero General Public License v3.0
516 stars 36 forks source link

FileNotFoundError: [Errno 2] No such file or directory: 'yolov8n.pt' #46

Closed huskycreatenode closed 6 days ago

huskycreatenode commented 1 week ago

你好，我想训练我的doclayout_yolo_ft.py模型 image 在windows 下，我可以正常的进行训练，但是在GPU服务器下，他总是会自动地去下载yolov8，我在train.py代码端没找到和yolov8有关的信息，我应该怎么解决？ image 我的训练命令为： python train.py --data d4la --model m-doclayout --pretrain /data/model/doclayout_yolo_ft.pt --device 0 --epoch 500 --image-size 1600 --batch-size 64 --project datasets/D4LA --plot 1 --optimizer SGD --lr0 0.04

huskycreatenode commented 1 week ago

image 应该是这个，所以CPU可以正常运行，我所疑惑的是为什么GPU要依赖于yolov8n.pt，这个amp_allclose(YOLO("yolov8n.pt"), im)的作用是什么呢？万分感谢解答

JulioZhao97 commented 1 week ago

您好：这个功能是原始ultralytics中自带的，您可以通过下载yolov8n.pt然后放置到项目根目录，或者通过以下解决方法：https://github.com/opendatalab/DocLayout-YOLO/issues/34