아하

생활

생활꿀팁

늘씬한박새60

늘씬한박새60

GAN 알고리즘으로 이미지를 생성하는법이 궁금합니다!

안녕하세요. 현재 컴퓨터공학부를 재학 중인 학부생입니다.

지금 MNIST 숫자 생성에 대해 공부를 마친 후 데이터 셋을 직접 생성하여

제가 원하는 연필, 지우개, 필통에 대한 이미지를 생성해보려 합니다.

이미지 크롤링을 하여 Label을 매기고, xml 파일을 생성하고, record와 csv를 생성을 하였습니다.

꽤 오랜 시간을 들여 하나하나 진행하고 있는데 무언가 잘못된 것을 느꼈습니다.

MNIST에서 (trainimages, trainlabels), (, ) = tf.keras.datasets.fashionmnist.loaddata()

loaddata() 부분에 들어가야 할 부분에 어떠한 데이터를 만들어야 할지 감이 잘 안 잡힙니다...

도움이 필요할 것 같습니다. ㅜ.ㅜ

어떤 식으로 GAN을 접근해야 할지도 알려주시면 감사하겠습니다!!

    2개의 답변이 있어요!

    • 탈퇴한 사용자

      탈퇴한 사용자

      tf.keras.datasets.fashionmnist.loaddata() 에서의 loaddata()는 단순히 fashionmnist의 데이터를 불러오기 위한 메소드입니다.

      따라서 커스텀 데이터를 불러오고 싶을 경우 opencv나 PIL 또는 tensorflow에서 지원하는 함수를 사용해야 합니다.

    • 냉엄한소쩍새267

      냉엄한소쩍새267

      우선 저 코드에서는 MNIST 데이터 파일을 불러오는 명령입니다. loaddata 함수를 포함하고 있어서 저렇게 불러서 쓰면 mnist 데이터 셋을 사용하는거라 본인이 만드신 학습 셋을 사용할수 없습니다. 파일 입출력을 통해서 csv 파일을 불러와서 train image train label로 넘겨주시고 학습을 하세요.