ascust / SEC-MXNet

MXNet implementation of SEC
MIT License
21 stars 4 forks source link

Seed Cue file contains blank arrays #16

Open karthik-sem opened 4 years ago

karthik-sem commented 4 years ago

I trained the model for generating the seed cues for my own dataset.But the foreground seed cue file contains blank arrays.Can you please suggest me how to correct it?

ascust commented 4 years ago

Sorry for the late reply. The foreground seed cues are basically generated based on the foreground saliency. SALIENCY_TH_FG is the parameter that controls whether a pixel should be selected as FG. you may have a look at this parameter.

Besides, the real reason for the blank arrays is still worth investigating. I recommend you make a n ablation study on your dataset, like removing those blank-array images etc. and see how it goes.

karthik-sem commented 4 years ago

Thank you @ascust .But how to we change the above parameter SALIENCY_TH_FG to select the desired pixel?

ascust commented 4 years ago

Just edit the parameter in "config.py" in the "core" folder. As I said, I would recommend doing some ablation study here by simply removing those blank-array images. This would be easier.