[일단 박죠] NVDIA의 GET3D를 활용한 아바타 자동 생성 #11

jeyong·2023년 10월 11일
0
post-custom-banner

더 많은 데이터를 활용해보기

지난번에 글에서 비교적 괜찮은 결과를 뽑았지만 아직 많이 부족한 상태이다. 데이터가 많이 부족하기 때문에 좋은 결과를 기대하기 어렵다. 그래서 더 많은 데이터를 구해서 활용해보기로 하였다. 인터넷을 돌아다니다가 데이터를 구할 수 있었다.

하지만 보는 것과 같이 배경에 흰색으로 입혀져있고 배경처리가 제대로 되어있지도 않고 폴더 구조도 이상했다. 해당 데이터를 아래 보이는 코드들을 이용해서 GET3D에 사용할 수 있도록 하였다.

학습

사진에서 보는 것과 같이 학습 해주었다

사진은 학습 옵션중 일부인데 무려 17352장의 데이터를 활용하였다.

해당 사진은 결과이다. kimg를 550까지 밖에 돌리지 못하였는데 이유는 컴퓨터의 성능때문이다.. 무려 3일간 돌렸으나 550까지 밖에 가지 못하였다.

그래서 컴퓨터가 학습하고 있는 사진을 확인하였는데 큰 이상을 찾지 못하였다.

550까지의 결과를 나타내느 사진이다. 굉장히 부족한 모습이다.

GET3D 논문을 읽어보았는데 GET3D측에서는 학습을 A100 GPU를 8대나 이용하여 학습하였는데도 이틀 걸렸다고한다... 다른 방안을 찾아봐야할 것 같다.

profile
노를 젓다 보면 언젠가는 물이 들어오겠지.
post-custom-banner

0개의 댓글