search

alanlukezic / d3s

D3S - Discriminative Single Shot Segmentation Tracker (CVPR 2020)
265 stars 57 forks source link

About test parameters #40

Open maojiaoli opened 2 years ago

maojiaoli commented 2 years ago

Hello, I used the trained model to test on THE GOT-10K, could you please tell me how I can get the visual effect of box and mask displayed at the same time? Where do you need to set the parameters? When I set debug = 1, all I can see is a visual display of the box.

alanlukezic commented 2 years ago

The current version of the code does not support visualization of the mask. You can save the masks to disk by setting the following parameters: params.save_mask = True, and params.masks_save_path = . The other way to display the masks more interactively is to modify the mask_to_disk.py so that it outputs the mask instead of returning it. To display the masks you will have to modify the code for visualization as well.