search

hackingmaterials / pv-vision

Image analysis tool for solar modules, assisted by deep learning.
BSD 3-Clause "New" or "Revised" License
35 stars 8 forks source link

How to obtain masks from raw images #13

Closed smttsp closed 2 months ago

smttsp commented 7 months ago

Hi there,

I was playing with the repo and I like the cleanliness and the quality of the code. Great job on that!

One issue I faced was that I couldn't figure out which model you were using to extract masks from raw images. I looked at all the notebooks but it is not clear to me at this point. I bet that it is module segmentation but I am not 100% sure.

I found pipeline.sh but it seems that it is mostly legacy work.

Could you please share a reference code if you have any available to perform that operation?

Thank you!

X-Chen97 commented 5 months ago

Hello!

Please confirm whether you want to extract masks of cracks, busbars etc. from the single cell images, or masks of a whole module from the tilted images.

This module segmentation allows to to extract module masks. But please note that our modules may not be the same as yours, so you will need to retrain that model using modelhandler in pv-vision. You can refer to this paper for the module segmentation

And for the crack segmentation, you can find the model weights here. You can refer to this paper for more information.

Thanks!

X-Chen97 commented 2 months ago

I will close this issue since there are no further questions asked. Feel free to raise new issues if you need help.