dataset

[ilous12]

@shurain 안녕하세요. 좋은 소스를 공개해주셔서 감사합니다.
이모델을 트레이닝 하려고 하는데, 데이터셋의 구성을 어떻게 해야 하는건가요?

제안하신 디렉토리 구조는 이해가 됩니다.

dataset_directory
|___ train
| |__ mask
| |__ image
|
|___ test
|__ mask
|__ image

이때 원본 이미지와 마스크 이미지의 형태를 어떻게 구성하면 될까요?

1. 이미지의 포맷은 어떻게 지정합니까? png, jpg 상관없나요?
2. 마스크의 영역의 구성을 확인하고자 합니다. 마스크는 grayscale 로 bg 0, shape 1 로 칠했는데, 맞을까요?

그리고 트레이닝이 끝나는 조건이 있나요? 너무 오래걸려서요

GPU Tesla K80 입니다.

image

mask

그리고, input / output node 명좀 먼지 알려주세요. freeze_graph 할려고 합니다. 혹시 covert_to_pb 설정이 가용한건지 확인도 부탁드립니다.

감사합니다.