train 시 augmentation으로 stain vector augmentation 또는 RandStainNA을 진행

4pygmalion commented 3 months ago

이 문제가 정말 스캐너 색감문제라면, 아마 SA(Stain augmentation)이 효과 있을거에요.

아래의 실험 부탁드릴 수 있을까요? @wjeong53

[x] : RandStainNA로 학습할때 augmentation을 하시구요. 확률P은 0.1~0.5정도로 임의의 float을 정하시면 될것같내요.
[x] : 브랜치 따서 작업하시고, 테스트 후에 효과가 있으면 RandStainNA을 git submodule로 넣으시면 될 것 같습니다.
[x] : Transform의 compose을 구성할 , PIL image인지 tensor인지 잘 확인하셔야합니다.
[x] : 모델은 experiments/train_ae.py을 이용해서 돌리시면되고
- [x] : alpha 값 0.3이 최적일 것 같내요.
- [x] : 모델: Unet 아키텍처
- [x] : 인코더 efficientnetb4
- [x] : 이미지 사이즈 -> 384

RandStainNA(2022?, 2023?): 리포가 공개되어있어요. RandStainNA++(2024년)써보셔도됩니다.

wjeong53 commented 3 months ago

네, 한번 작업해보겠습니다

4pygmalion commented 2 months ago

RandStainNA 작업 진행중 (https://github.com/4pygmalion/cosas/issues/102) => segformer에 학습중

4pygmalion / cosas