search

mxin262 / SwinTextSpotter

Pytorch re-implementation of Paper: SwinTextSpotter: Scene Text Spotting via Better Synergy between Text Detection and Text Recognition (CVPR 2022)
267 stars 40 forks source link

关于SwinTextSpotter and ESTextSpotter 一些问题 #107

Open Micla-SHL opened 11 months ago

Micla-SHL commented 11 months ago

作者你好，我是了解到ABCNet后查找到最新的SwinTextSpotter项目，我觉得它应该是比ABCNet更优秀，您在SwinTextSpotter提到更新的ESTextSpotter，我在昨天尝试之后，能 执行 vis.py （需在for循环末尾添加: torch.cuda.empty_cache()，) 我的GPU都是独立的，3060,3070,4090。单张GPU资源能否支持这两个项目之一，我的资源太少了，没有8张。我即使GPU跑一张图显存也不够(在3060实验ESTextSpotter，还未测试4090,)。这两项目对资源的需求能否再下降? 这是可行的吗? 如果我对网络不管精度先剪枝确保能训练这是能被推荐的吗? 还是我这边添加设备会更好些?

Micla-SHL commented 11 months ago

更新资源消耗: 4090 显存24G 运行bash scripts/Pretrain_Micla.sh /Micla/Project/ESTextSpotter/datasets/ datasets只包含totaltext数据集，Pretrain_Micla.sh 相比原版作了显卡8>1 train_dataset剔除剩下 totaltext ，配置batch size 2> 1, 所需显存 21G

mxin262 commented 10 months ago

可以把ESTextSpotter中的denoising训练部分关了可以节省显存

Micla-SHL commented 10 months ago

好的，我去尝试一下。

发件人: mxin262 @.> 发送时间: 2023年10月23日 14:56 收件人: mxin262/SwinTextSpotter @.> 抄送: Micla-SHL @.>; Author @.> 主题: Re: [mxin262/SwinTextSpotter] 关于SwinTextSpotter and ESTextSpotter 一些问题 (Issue #107)

可以把ESTextSpotter中的denoising训练部分关了可以节省显存

― Reply to this email directly, view it on GitHubhttps://github.com/mxin262/SwinTextSpotter/issues/107#issuecomment-1774543915, or unsubscribehttps://github.com/notifications/unsubscribe-auth/AKTQYIVXXYYGZXMCHHU4GYDYAYIINAVCNFSM6AAAAAA55CYQU2VHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMYTONZUGU2DGOJRGU. You are receiving this because you authored the thread.Message ID: @.***>