search

dawn-ech / YOLC

[TITS 2024] You Only Look Clusters for Tiny Object Detection in Aerial Images
https://arxiv.org/abs/2404.06180
Apache License 2.0
72 stars 13 forks source link

请问能开源权重吗？ #11

Open 2225686820 opened 2 months ago

2225686820 commented 2 months ago

作者您好，我尝试用单张3090跑这个模型的训练，发现用时很长，整个流程训练下来大概需要一周，请问这正常吗？另外请问您是否可以开源您训练好的权重呢？

2225686820 commented 2 months ago

另外复现过程中我发现好像在label转换过程中的代码不是很完整，缺少val的label生成部分

dawn-ech commented 2 months ago

确实训练是久一点，4卡3090大约需要2天。val生成的原始脚本找不到了，我稍后上传一个转换好的json吧。权重我后续再上传

Rrenxy commented 2 months ago

作者您好，同求训练好的权重文件

dawn-ech commented 1 month ago

权重已上传，可以在README中查看