foolwood / SiamMask

[CVPR2019] Fast Online Object Tracking and Segmentation: A Unifying Approach
http://www.robots.ox.ac.uk/~qwang/SiamMask
MIT License
3.47k stars 816 forks source link

simple question about the input of the model #154

Open Ryan315 opened 4 years ago

Ryan315 commented 4 years ago

I have a simple question. How the input is organized? I saw the input is

   x = {

'cfg': cfg, 'template': torch.autograd.Variable(input[0]).cuda(), 'search': torch.autograd.Variable(input[1]).cuda(), 'label_cls': torch.autograd.Variable(input[2]).cuda(), 'label_loc': torch.autograd.Variable(input[3]).cuda(), 'label_loc_weight': torch.autograd.Variable(input[4]).cuda(), 'label_mask': torch.autograd.Variable(input[6]).cuda(), 'label_mask_weight': torch.autograd.Variable(input[7]).cuda(), } but I didn't find related code in 'siam_mask_dataaset.py' .

Ryan315 commented 4 years ago

...ok I've found them.