hellochick / PSPNet-tensorflow

TensorFlow-based implementation of "Pyramid Scene Parsing Network".
326 stars 123 forks source link

The auxiliary loss #41

Open Ariel-JUAN opened 6 years ago

Ariel-JUAN commented 6 years ago

Apart from the main branch using softmax loss to train the final classifier, another classifier is applied after the fourth stage, i.e., the res4b22 residue block. Do anyone know how to write the auxiliary loss code?

gaowu9595 commented 5 years ago

I wonder, too!