hyperconnect / MMNet

Code for Towards Real-Time Automatic Portrait Matting on Mobile Devices
https://arxiv.org/abs/1904.03816v1
Apache License 2.0
177 stars 36 forks source link

dataset #6

Open ilous12 opened 5 years ago

ilous12 commented 5 years ago

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

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

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

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

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

GPU Tesla K80 입니다.

image 00001_top

mask 00001_top

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

감사합니다.