official measure code about osvos-pytorch

[kmaninis]

Yes, both are binary. The optimal threshold is around 0.8.

[InstantWindy]

But when I set the threshold of 0.8, the iou is 0,I don't know why .My code is as follows:

[scaelles]

Hello,
The code that you provide seems right to me. I would do a couple of things to double check that all the tensor have the appropriate values:

• Check that probabilities and gts are correct: np.unique(probabilities.cpu().numpy()) should only be 0 and 1.
• Visualize with matplotlib the probabilities tensor before thresholding.

[InstantWindy]

Yeah,Maybe it's because I trained on the Davis 2017 dataset， Davis 2017 dataset labels have multiple categories.If I want to train on the Davis 2017 dataset, how do I set the label values of multiple categories to 0 and 1? My idea is to set the pixel values of the target category to 1 and the background to 0.

[scaelles]

In the multiple object scenario, the evaluation gets a little bit more tricky so I would recommend you using the python package that we provide for DAVIS 2017. If there are multiple objects, each object pixel value should be the same as the one for that object in the first frame.

[InstantWindy]

I use the evaluation code provided by the official davis2017. But this evaluation code requires the input to be a binary map. So how do you convert multi-object labels into binary maps? Thank you!

[scaelles]

I think that the best is that you evaluate the whole sequence and not every frame using this.

In case that you want to evaluate per frame, you have to create yourself a loop evaluating one object at a time. Keep in mind that the DAVIS evaluation is done obtaining the mean of every object in the whole sequence and then doing the mean of all the objects. So don't do the mean of all the objects in each frame.