강성준
강성준
🎻 루키 파트너
🌈 지피터스금손

[허깅페이스x내집마련] 부동산 관련 이미지 자동 분류기 Real estate image classifier model 만들기(Feat. HuggingFace)

안녕하세요.

허깅페이스 캠프 11기 서포터이자, 내집마련 청강생 강성준입니다.

본업은 바이오텍 신약개발입니다만, 개인적으로 AI(AI신약개발)에도 관심이 있고, 하고 싶은거 자유롭게 하면서 살려면 경제적으로도 독립성이 보장되어야 겠다는 생각에, 투자 공부도 좀 해보고자 청강 신청하였습니다!

오늘은 허깅페이스방 출신 답게 허깅페이스에서 무언가 부동산과 관련한 재미난 일을 해볼수 있지 않을까 하는 마음에

부동산 매물의 이미지를 보고 어떤 종류인지 분류해내는 AI모델을 우리 허깅이의 도움으로 한번 만들어 보았습니다.

(담주 허깅페이스 발표용이 메인이지만, 내집마련 청강발표에도 사용해보려고 합니다..!ㅋㅋㅋㅋ)

결과물부터 보시겠습니다.

Kang-Seong-Jun/GPTers_11th_Real_Estate_Classifier · Hugging Face

오늘 학습시킨 따끈따끈한 아기모델(응애)입니다.

아파트, 단독주택, 거실, 주방, 침실, 서재, 사무실, 화장실 등등의 구분이 가능합니다.

아주 기본적인 분류가 가능하지만 더 발전시킨다면, 여러가지 재미있는 활용이 가능하지 않을까 하는 마음입니다!
(금액대 추론 등등…? 사실 제가 찐 부린이라 어떻게 발전시켜서 어떻게 활용할지 아이디어 있으시면 많은 지도편달 부탁드리겠습니다!)

학습시킨 이미지는 보시면은 살짝~ 서양느낌이 있습니다만! 뒤에 보시면 한국 매물이미지도 잘 구분해 냅니다!!

아래는 실제로 HuggingFace Space에서 실제 한국 집 사진을 가지고 모델 돌린 모습입니다!!

URL로 들어가시면 실제로 직접 사용도 가능하십니다.(좋아요도 많이 눌러주세요)
GPTers Real Estate Classifier - a Hugging Face Space by Kang-Seong-Jun

처음 들어가시면, 아래와 같이

이미지를 넣는 input image 칸(사진을 직접 넣을수도 있고, 복사해놓고 클립보드 붙여넣기도 됩니다!)와 옆에 Classification 창이 보입니다! 사진을 넣으면 결과를 보여주는 창이 되겠습니다.

아래는 인터넷에 있는 실제 한국의 아파트, 내부, 외부 사진들로 테스트해본 결과들입니다!

한국 아파트 거실(Living room)은 물론이고,

화장실(Bathroom)도 바로 캐치해냅니다.

체리몰딩이 매력적인 한국식 주방도 잘 분류해 내는군요!

“아파트”와 “단독주택” 역시도 정확하게 분류해내는 모습!!!

직접 한번 아래에 들어가서, 갖고 계신 부동산 사진으로 한번 테스트 해보시면 좋을것같습니다!! (당일에 뚝딱만든 응애 모델이기때문에, 정확하지 않더라도 넓은 아량을 보여주세요..!ㅎㅎ아주 복잡한 매물이나, 어려운 사진 구분은 아직 조금 부족하답니다!)

GPTers Real Estate Classifier - a Hugging Face Space by Kang-Seong-Jun

사람이 분류하는것과 무엇이 다른가 싶긴 하지만, 이런 소소한 모델들이 모이고 모이면 나중에 자동화와 대량 분류를 통해서 뭔가 재밌는 활용법이 있지 않을까 하는 작은 상상을 해봅니다!!

“허깅페이스 x 내집마련” 화이팅!!!

#11기내집마련 #11기HuggingFace

5
7개의 답글

👉 이 게시글도 읽어보세요