eqlv2 icon indicating copy to clipboard operation
eqlv2 copied to clipboard

About baseline model

Open U-Help opened this issue 3 years ago • 4 comments

I have great interest in your work, and I want to ask a question. Does the baseline model also use sigmoid loss function rather than softmax loss function?

U-Help avatar Mar 24 '21 12:03 U-Help

Thanks for your interest. In this repo, the baseline model is trained with softmax loss function.

tztztztztz avatar Mar 25 '21 05:03 tztztztztz

Thanks for your reply. In your paper, at this location, you refer baseline model to sigmoid loss function, image whereas at corresponding picture, it becomes softmax loss function. I'm a little confused. image

U-Help avatar Mar 25 '21 07:03 U-Help

Thanks for pointing out this typo. In Figure 2, we use the baseline models with the sigmoid loss function.
We have already corrected it in the camera-ready version, which will be updated to arxiv soon.

tztztztztz avatar Mar 26 '21 02:03 tztztztztz

Hi Authors, can this work be combined into YOLO v5 ?

ShirleyHe2020 avatar Mar 31 '21 02:03 ShirleyHe2020