[Reproduce] Add VisDrone & SKU-110K training scripts and full results - #113
Open
Alan-20251210 wants to merge 2 commits into
Open
[Reproduce] Add VisDrone & SKU-110K training scripts and full results#113Alan-20251210 wants to merge 2 commits into
Alan-20251210 wants to merge 2 commits into
Conversation
新增VisDrone与SKU-110K数据集复现训练脚本: 1. reproduce_visdrone.py:针对VisDrone数据集的模型训练、评估完整流程 2. reproduce_sku110k.py:SKU-110K商品检测数据集训练脚本 适配当前分支的实验复现流程,可直接运行训练
1. Add reproduce training scripts for VisDrone and SKU-110K datasets, support v0.1-N baseline and EsMoE-N model, compatible with Windows multi-process training. 2. Create README_reproduce.md with complete usage guide: dataset download steps, training commands, quantitative mAP comparison table, offline WandB log download link and environment configuration. 3. Remove large wandb zip packages from repository, store offline logs on Baidu Netdisk to avoid bloating repo storage. 4. Supplement known training issues, hardware environment parameters and dependency installation instructions for 8GB laptop GPU users.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Closes #49
犀牛鸟任务:VisDrone & SKU-110K 复现结果
本 PR 新增了 VisDrone 和 SKU-110K 数据集上的训练脚本及完整复现结果(离线日志提供百度网盘下载)。
新增文件
scripts/reproduce/reproduce_visdrone.py:支持--model v01 / moe,适配 Windows 多进程scripts/reproduce/reproduce_sku110k.py:同上README_reproduce.md:详细说明、训练命令、结果表格及百度网盘离线日志下载链接复现结果
训练日志
README_reproduce.md中提供永久下载链接及提取码,解压后wandb sync即可查看全部训练曲线。tiw2本地复现命令
环境配置
已知问题说明
workers=0,否则可能引发 I/O 死锁。imgsz=800为推荐最低值,降低分辨率可能导致精度下降。