[Model] CascadeEncoderDecoder

boostcampaitech3 / level2-semantic-segmentation-level2-cv-09

level2-semantic-segmentation-level2-cv-09 created by GitHub Classroom

2 stars 4 forks source link

Closed km9mn closed 2 years ago

km9mn commented 2 years ago

mmseg의 model type에는 EncoderDecoder 말고도 CascadeEncoderDecoder가 있다 https://github.com/open-mmlab/mmsegmentation/blob/master/mmseg/models/segmentors/cascade_encoder_decoder.py
CascadeEncoderDecoder로 실험
decode_head의 output이 다음 decode_head의 input으로 들어가는 방식
Swin-L 로 실험

km9mn commented 2 years ago

mmseg에는 ocrnet과 pointrend 모델이 CascadeEncoderDecoder를 쓰고있다. https://github.com/open-mmlab/mmsegmentation/search?p=1&q=cascade ocr은 fcn-ocr, pointrend는 fcn-point로 cascade stage가 구성되어있다.

km9mn commented 2 years ago

좀더 서치해봐야겠지만 object detection과는 달리 cascade가 같은 head를 반복하기 보다는 서로다른 head를 이어붙이는 느낌

km9mn commented 2 years ago

OCRNet의 CascadeEncoderDecoder는 잘돌아가지만 나의 Cascade는

AssertionError: CascadeEncoderDecoder: OCRHead:

이런 에러를 뿜는다..