Deeplab v3 turf datasets 훈련

BERT·2023년 5월 19일
0

Segmentation

목록 보기
8/10

데이터 폴더 생성

데이터 선정

훈련에 적합하다고 판단되는 데이터 20장 선정 후 훈련 데이터 폴더에 복사

훈련 : 검증 = 0.85 : 0.15 = 17 : 3

훈련

Backbone ModelResnet50
Image Shape(224 x 224 x 3)
Mask Shape(224 x 224 x 2)
Number of output channels2
(0 – background, 1 – abnormal)
Number of Epochs Trained50
Batch Size3
OptimizerAdam
Learning Rate0.0001
Loss Function/sBinary Cross entropy
Intersection over union
(BCE + IoU)
Evaluation MetricIntersection over Union (IoU)
python main_train.py --epochs 50 --batch 3

검증

python main_test.py --input ../input/Turf_Dataset/valid_images/
ori_datavalid_data
image.jpg1image.jpg2

.

ori_datavalid_data
image.jpg1image.jpg2

테스트

python main_test.py --input ../input/Turf_Dataset/test_images/

참조

Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation

0개의 댓글