YOLOv6 icon indicating copy to clipboard operation
YOLOv6 copied to clipboard

yolov6的0.3.0版本能够导出torchscript模型么

Open arkerman opened this issue 1 year ago • 3 comments

Before Asking

  • [X] I have read the README carefully. 我已经仔细阅读了README上的操作指引。

  • [ ] I want to train my custom dataset, and I have read the tutorials for training your custom data carefully and organize my dataset correctly; (FYI: We recommand you to apply the config files of xx_finetune.py.) 我想训练自定义数据集,我已经仔细阅读了训练自定义数据的教程,以及按照正确的目录结构存放数据集。(FYI: 我们推荐使用xx_finetune.py等配置文件训练自定义数据集。)

  • [X] I have pulled the latest code of main branch to run again and the problem still existed. 我已经拉取了主分支上最新的代码,重新运行之后,问题仍不能解决。

Search before asking

  • [X] I have searched the YOLOv6 issues and found no similar questions.

Question

yolov6的0.3.0版本能够导出torchscript模型么

Additional

No response

arkerman avatar Nov 02 '23 09:11 arkerman

可以。

mtjhl avatar Nov 13 '23 02:11 mtjhl

但是您需要实现一下,我们后续也会加入这个代码,代码很简单。

mtjhl avatar Nov 13 '23 02:11 mtjhl

Do u have the code now to convert it to torchscript? Better if can set the image size as well when exporting.

evertale888 avatar Dec 13 '23 18:12 evertale888