-
trt命令
`trtexec --onnx=/data/DocLayout_yolo/models/doclayout_yolo_docstructbench_imgsz1024.onnx --saveEngine=/data/DocLayout_yolo/models/model_best.plan --workspace=300000 --device=0 --noTF32 --v…
-
### Search before asking
- [X] I have searched the Ultralytics YOLO [issues](https://github.com/ultralytics/ultralytics/issues) and [discussions](https://github.com/ultralytics/ultralytics/discussion…
-
新建的版面分析模型doclayout 输入shape是固定的1024x1024, 这样会有两个问题：
1、有的图片会因非等比例缩放， 导致结果不准
2、增加计算量
建议， 将onnx模型的输入shape转成dynamic, 图像使用letterBox预处理后送入模型
_Originally posted by @jesse01 in https://github.com/RapidAI…
-
你好 有没有相关的文本版面分析的数据介绍，我看paddle并没有写相关数据介绍
-
## 问题描述
FutureWarning: You are using `torch.load` with `weights_only=False` (the current default value), which uses the default pickle module implicitly. It is possible to construct malicious pickle …
-
运行的脚本
D4LA_finetune(){
python train.py \
--data d4la \
--model m-doclayout \
--epoch 1 \
--image-size 1600 \
--batch-size 16 \
--project No…
-
你好 我用自己的数据进行训练 参考了你的配置
python train.py --data doclaynet --model m-doclayout --epoch 500 --image-size 1120 --batch-size 64 --project public_dataset/doclaynet --plot 1 --optimizer SGD --lr0 0.02
发现mod…
-
I have downloaded the dataset of DocLayNet and tried to initiate the train.py I am getting this error :
Please help me with this .
ckpt = torch.load(file, map_location="cpu")
Traceback (most …
-
Hi. Thanks for developing such a good model.
Inspired by your model, I'm trying to train a CustomDataset using DocLayout-YOLO, but I'm getting the following error.
` ckpt = torch.load(file, map_l…
-
你好，我想训练我的doclayout_yolo_ft.py模型
在windows 下，我可以正常的进行训练，但是在GPU服务器下，他总是会自动地去下载yolov8，我在train.py代码端没找到和yolov8有关的…