search

MichaelFan01 / STDC-Seg

Source Code of our CVPR2021 paper "Rethinking BiSeNet For Real-time Semantic Segmentation"
MIT License
703 stars 131 forks source link

STDC backbone for imagenet #73

Closed jjxxmiin closed 2 years ago

jjxxmiin commented 2 years ago

Thanks for the good work.

How to train STDC backbone on ImageNet?

I think the STDC backbone consists of Stage 1 to Stage 5.

Then, is the STDC backbone for the imagenet created by attaching a fully connected layer to the last global average pooling?

Asthestarsfalll commented 2 years ago

https://github.com/MichaelFan01/STDC-Seg/blob/master/nets/stdcnet.py#L195 I think it's for ImageNet classification.

jjxxmiin commented 2 years ago

Many Thanks!!

I understood it.