chineseocr_lite icon indicating copy to clipboard operation
chineseocr_lite copied to clipboard

Published 20 hours ago • verified publisher icon DayBreak-u

能提供dbnet的训练代码和onnx转换代码吗

Open 2017ZYS opened this issue 4 years ago • 15 comments

2017ZYS avatar Jul 13 '20 02:07 2017ZYS

https://github.com/WenmuZhou/DBNet.pytorch

DayBreak-u avatar Jul 13 '20 08:07 DayBreak-u

您好，项目里面DBNet用的什么backbone呢？

deep-practice avatar Jul 14 '20 08:07 deep-practice

mobilenetv3

2017ZYS avatar Jul 14 '20 12:07 2017ZYS

@ZetaoYang 好的，谢谢

deep-practice avatar Jul 14 '20 14:07 deep-practice

请问 mobilenetv3 训练出来的原始模型，大小是多少啊？ @2017ZYS

Tian14267 avatar Jul 20 '20 07:07 Tian14267

如果是mv3-large0.5，是3.06MB,详细看https://github.com/WenmuZhou/DBNet.pytorch/blob/master/models/backbone/MobilenetV3.py

mobilnetv3 = MobileNetV3(in_channels=3,scale=0.5,model_name="large") torch.save(mobilnetv3.state_dict(), 'mobilnetv3.pth')

2017ZYS avatar Jul 20 '20 11:07 2017ZYS

好的，多谢 @2017ZYS

Tian14267 avatar Jul 21 '20 01:07 Tian14267

请问训练采用的数据集是哪个数据集？

fengxingxiang avatar Jul 22 '20 02:07 fengxingxiang

@fengsong77 lsvt

DayBreak-u avatar Jul 22 '20 02:07 DayBreak-u

@ouyanghuiyu 谢谢大佬

fengxingxiang avatar Jul 23 '20 02:07 fengxingxiang

@ouyanghuiyu 大佬，项目里面的dbnet backbone是基于mobilnetv3做了修改吗

lgcy avatar Sep 04 '20 10:09 lgcy

@ouyanghuiyu 大佬，项目里面的dbnet backbone是基于mobilnetv3做了修改吗

基本没改的

DayBreak-u avatar Sep 05 '20 07:09 DayBreak-u

大佬，请问dbnet网络都做了哪些简化？我用工程https://github.com/WenmuZhou/DBNet.pytorch 中MobileNetV3训练出来比你提供的要慢许多，请问是不是neck和head网络做了简化，另外还有什么加速优化吗？谢谢！

reason4 avatar Nov 02 '20 09:11 reason4

@ouyanghuiyu 大佬，crnn使用的是什么backbone呀？

zhangmytf avatar Jul 07 '21 02:07 zhangmytf

如果是mv3-large0.5，是3.06MB,详细看https://github.com/WenmuZhou/DBNet.pytorch/blob/master/models/backbone/MobilenetV3.py

mobilnetv3 = MobileNetV3(in_channels=3,scale=0.5,model_name="large") torch.save(mobilnetv3.state_dict(), 'mobilnetv3.pth')

为啥我的是29MB，作者的backbone替换成了mbv3，其它没动，大佬救命

xspring14 avatar Oct 13 '22 09:10 xspring14