CV-16조 💡 비전길잡이 💡
NAVER Connect Foundation boostcamp AI Tech 4th
민기 | 박민지 | 유영준 | 장지훈 | 최동혁 |
---|
- 비전 길잡이 Notion 📝( not finished )
- weighted_sum head에 대한 성능 검증
ADE20K | |
---|---|
Purpose | Fine-tuning |
Num_classes | 150 |
Training set | 20,210 images |
Validation set | 2,000 images |
|-- ADEChallengeData2016
|-- image
| |-- train
| `-- val
`-- mask
|-- train
`-- val
git clone https://github.com/revanZX/mmseg_WS.git
# with WS
python tools/train.py \
./configs_custom/pvt_v2/fpnws_pvt_v2_b5_ade20k_160k.py \
--work-dir ../work_dir/pvtws_v2_b5 \
--seed 1
# with NON WS
python tools/train.py \
./configs_custom/pvt_v2/fpn_pvt_v2_b5_ade20k_160k.py \
--work-dir ../work_dir/pvt_v2_b5 \
--seed 1
|-- 🗂 appendix : 발표자료 및 WrapUpReport
|-- 🗂 segformer : HuggingFace 기반 segformer 모델 코드
|-- 🗂 boostformer : Segformer 경량화 모델 코드
|-- 🗂 imagenet_pretrain : Tiny-ImageNet encoder 학습시 사용한 코드
|-- 🗂 util : tools 코드 모음
|-- Dockerfile
|-- train.py : ADE20K Finetuning 코드
|-- eval.py : 모델 Inference 결과 출력 코드
|-- requirements.txt
`-- README.md