alibaba / alibaba/esod

第一个训练指令已经是使用ESOD了吗,评估结果有gap

Open
#11 6 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
140
Forks
21
PR merge metrics
No merged PRs in 30d

Description

如题,训练指令为DATASET=visdrone MODEL=yolov5m GPUS=2 BATCH_SIZE=8 IMAGE_SIZE=1536 EPOCHS=50 bash ./scripts/train.sh
评估指令为python test.py --data data/visdrone.yaml --weights runs/train/exp2/weights/best.pt --img-size 1536 --device 0,1
只评估了visdrone数据集,评估输出结果如下图

![Image](https://github.com/user-attachments/assets/c3e71959-5967-49e9-8e8b-18b768486156)

显示map只有34.5,与论文的36.0有点差距,是我哪里设置的问题吗,还麻烦作者帮忙看看,感谢!

Contributor guide

No contributing guide indexed for this repository

Research direction

Reproduce the stated training command from scripts/train.sh and the evaluation command in test.py using data/visdrone.yaml and the reported weights path. Compare the resulting mAP with the attached 34.5 result and the paper's 36.0, then inspect the training and evaluation settings to identify the mismatch. Done means explaining the gap or confirming the settings needed to match the paper.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, shell
Domain
computer-vision, machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.