Я новичок в Tensorflow, я тренировал модель обнаружения объектов на основе SSD, и у меня возникли проблемы с перекосом классов переднего плана и фона. Я попробовал потерю очага сигмовидной кишки, но это, похоже, не слишком помогает. У меня 500 000 аннотаций, перекошенных для нескольких топ-классов.
Я считаю, что реализация потерь GHM поможет моему делу. Я смотрел на object_detection / core / loss.py в репозитории моделей Tensorflow, но не знал, как реализовать функцию потерь. Любые указатели помогут.
Вот некоторые подробности, которые я нашел в другом репо для GHM https://github.com/xyfZzz/GHM_Loss_Tensorflow/blob/master/ghmc_tensorflow.py