关于 vanilla 与 ESOD 版本推理速度差异的疑问 #21
Description
您好，
我在相同环境下分别训练并测试了 YOLOv5 和 RetinaNet 的 ESOD 版本 与 vanilla 版本，结果发现无论是哪种模型，vanilla 版本的推理速度都显著快于 ESOD 版本。
在输入图像尺寸为 1536×1536 的情况下，测试得到的 FPS 如下：
我训练模型使用的命令为DATASET=visdrone MODEL=yolov5m（或retinanet） GPUS=0 BATCH_SIZE=8 IMAGE_SIZE=1536 EPOCHS=50 bash ./scripts/train.sh，并除了将train.sh中的model_type进行esod/vanilla的调整以外没有对代码部分进行任何修改。
我测试fps使用的命令为python test.py --data data/visdrone.yaml --weights runs/train/expn/weights/best.pt --batch-size 1 --img-size 1536 --device 0 --task measure。
请问是我的训练、测试过程中有什么地方有错误吗？如有误的话该如何调整以获取和论文结果相近的推理速度结果呢。