- 在yolox正添加训练yolov5的数据格式,以及测试代码。
- 目前是在单gpu模式下调试的代码, 多gpu下还需要修改。
- 具体修改:
- 在yolox/data/datasets/下添加yolo.py和yolo_classes.py
- 在yolox/data/datasets/init.py中添加
- from .yolo import YOLODataset
- from .yolo_classes import YOLO_CLASSES
- 在yolox/evaluators/下添加 yolo_evaluator.py
- 在yolox/evaluators/init.py中添加:
- from .yolo_evaluator import YOLOEvaluator
- 训练的配置文件参考yolox_yolo_s.py
- 修改代码, 多gpu下分别计算各自卡上的map,未做多gpu同步计算。
- 增加多gpu计算map的代码
xialuxi/yolox-yolov5
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published