search

ai-starthon / AI_Starthon2019

60 stars 44 forks source link

7_icls_face 의 inference 부분 #153

Open stimong opened 5 years ago

stimong commented 5 years ago

안녕하세요. keras예제 참고해서 keras작성중에 inference 부분때문에 질문드립니다. inference 부분에서 테스트 데이터셋을 어떻게 불러들여야할지.. main.py에서 training부분은 TRAIN_DATASET_PATH에 작성되어있는데

> 
> try:
>     import nsml
>     from nsml import DATASET_PATH, IS_ON_NSML
>     TRAIN_DATASET_PATH = os.path.join(DATASET_PATH, 'train', 'train_data')
>     VAL_DATASET_PATH = None
> except:
>     IS_ON_NSML=False
>     TRAIN_DATASET_PATH = os.path.join('/media/yoo/Data/NIPAFaceCls/train/train_data')
>     VAL_DATASET_PATH = None
> ....
>

VAL_DATASET_PATH 또는 test데이터셋은 경로를 알수없어서, 어떤식으로 inference에 작성해야할지 모르겟습니다. test데이터셋 경로나, 읽는방법, 최종 output형식 등 정보 부탁드립니다.

nsml-admin commented 5 years ago

안녕하세요.

VAL_DATASET_PATH 는 문제마다 따로 존재하는경우도 있고 없는경우도 있습니다. (이부분은 베이스라인을 잘 살펴보시면 될것같습니다.) 없는경우는 train위치경로에서 나눠서 사용하시면 될듯합니다.

test셋의 위치는

TEST_DATASET_PATH = os.path.join(DATASET_PATH, 'test')

이런식으로 가져올수있습니다. (https://github.com/ai-starthon/AI_Starthon2019/blob/master/7_icls_face/main.py#L57-L59)

데이터셋을 읽는 방법은 @stimong 님이 어떤 모듈을 사용해서 읽을지에 따라서 다르므로 이부분은 직접 작성하셔야 합니다(이부분또한 각 baseline에 실제 돌아가는 예제로 작성되어 있습니다. data_loader.py부분을 참고하시면 될듯합니다.)

최종 output형식 역시 마찬가지로 각 문제마다 infer에서 반환하는 format에 맞춰서 작성하시면 됩니다. (참고로 7번문제는 여기 입니다.)

감사합니다.

stimong commented 5 years ago

@nsml-admin 제가 DATASET_PATH: /data/7_icls_face 하위 폴더 명 추출했더니 ['train'] 만 출력되었습니다.

/data/7_icls_face/test 의 폴더 존재여부 확인했더니 false로 출력됩니다.