TIL 2023.06.08

오늘은 어제 만든 인물 segmetation 기능을 조금 손보았다.

 

먼저 모델을 교체했다.

DeepLabV3를 썼던 것을 FCN으로 바꿔주었다

여기서 알게 된 사실이

torchvision의 segmentation model은 이미지를 PIL.Image혹은 torch.Tensor의 형식만 받는 다는 것이었다.

결과 반환도 Tensor 형식이기에 openCV를 이용해 편집하려면 Numpy 형식으로 변환을 해줘야 했다.

detach().cpu().numpy() 명령어로 변환을 해줄 수 있었다.

 

그리고 배경 합성부분까지 작성을 했다.

 

다른 세그멘테이션 모델도 찾아보다가

https://youtu.be/S54EprtQdjA

이런 영상도 봤는데 더 최신 모델인 것 같다. 시간이 덜 걸린다면 이걸 쓰는 게 나을 수도

 

https://www.kogl.or.kr/index.do

공공누리

배경으로 쓸 사진을 찾다가 이런 사이트도 발견했다.

출처만 밝히면 영리 목적으로도 쓸 수 있는 사진들이 많다.