checkpoint of the model trained LVISv0.5 + COCO dataset

[gaoyuchris]

Hi, I wonder can you provide the checkpoint of the model trained on LVISv0.5 + COCO dataset? many thanks.

[siyuanliii]

Thanks. For fintuning on TAO, we freezed the detection parts and only train the track head. Thus, you can download the weights in the repo and drop track head part to get the model that you want.

[gaoyuchris]

Thanks. For fintuning on TAO, we freezed the detection parts and only train the track head. Thus, you can download the weights in the repo and drop track head part to get the model that you want.

Thanks for your reply @siyliepfl . I understand that the track head will also be trained in the training on LVISv0.5 + COCO dataset, which take the key frame itself as its ref frame (according to

tet/configs/tao/cem_r101_lvis.py

Line 31 in 2c18172

track_head=dict(

). Thus the weights in the repo with the track head part dropped are not the weights I want.