i-pan/kaggle-siim-ptx

Avoid error: index 0 is out of bounds for axis 0 with size 0 when exporting png maks

casfranco opened this issue · 0 comments

On file "etl/1_get_png_masks_and_assign_labels.py", line 41 there is an extra space that is causing an error when exporting the .png masks.
This happened when using the current kaggle dataset
https://www.kaggle.com/seesee/siim-train-test

-->
if df[' EncodedPixels'].iloc[0] == '-1':

Correcting that line the code execute and export the mask,