There is an error in the code
Closed this issue · 1 comments
wangyujie615 commented
File "/data2/wangyujie/ZoomTrack/lib/train/../../lib/train/actors/hivitr_zoom_actor.py", line 84, in compute_losses
pred_boxes_vec = unwarp_bboxes_batch(box_cxcywh_to_xyxy(pred_dict['pred_boxes']), gt_dict['search_grids'][0]).view(-1, 4)
KeyError: 'search_grids'
Kou-99 commented
- check if you use QpProcessing (lib.train.data.processing.py) as data processing
- check if arg use_grid is correctly configured when initializing QpProcessing
An example can be found in /lib/train/train_settings/ostrack.py line 73~95