search

PaddlePaddle / PaddleDetection

Object Detection toolkit based on PaddlePaddle. It supports object detection, instance segmentation, multiple object tracking and real-time multi-person keypoint detection.
Apache License 2.0
12.65k stars 2.87k forks source link

如何用atss替换yolox中的simota？ #6444

Closed ZiQiangXie closed 2 years ago

ZiQiangXie commented 2 years ago

问题确认 Search before asking

请提出你的问题 Please ask your question

如何用atss替换yolox中的simota？

nemonameless commented 2 years ago

简单起见可以把ppyoloe head整体替换yolox head，static_assigner_epoch设置为最大epoch数就是全程ATSS，把reader也可以换下。 如果单独只换simota那一步改动稍微麻烦点，需要保持接口参数一致。但simota已经证明在coco上高于ATSS，在大部分数据集上也一样，个人觉得不是很有必要替换。

ZiQiangXie commented 2 years ago

感谢回复。看了接口，确实修改会比较麻烦一些。想替换主要是因为simota训练比较慢，想换成ppyoloe那种分配方式，但是又需要用yolox的检测头。