(Keras) keras segmentation model API

segmentation문제 중에

 

1. Custom model을 하기 귀찮거나,

2. 적절한 model을 우선적으로 찾은 후 그 모델을 customizing하기 위한 방법으로

 

keras를 기반으로하는 API를 활용해서 빠르게 segmentation 모델을 적용하는 것입니다.

 

현재, 꼼꼼히 찾아보진 못했지만 2가지 API가 잘 되어 있는 것 같습니다.

 

물론, pip install을 활용해서 install을 해야합니다.

 

* tensorflow 2.1버전 이하로만 가능합니다. 즉, kears 2.4버전 이하

 

1. segmentation_model

segmentation-models.readthedocs.io/en/latest/index.html

Welcome to Segmentation Models’s documentation! — Segmentation Models 0.1.2 documentation

github : github.com/qubvel/segmentation_models

qubvel/segmentation_models

 

위는 API에 관하여 문서화도 되어 있어서, 저는 1번을 사용합니다.

 

2. keras_segmentation

github.com/divamgupta/image-segmentation-keras

divamgupta/image-segmentation-keras

 

2가지를 본인의 입맛에 맞게 활용하면, 빠르게 modeling후 본인이 풀고자하는 문제에 적절한 model을 찾을 수 있을 것 같습니다.