tet icon indicating copy to clipboard operation
tet copied to clipboard

checkpoint of the model trained LVISv0.5 + COCO dataset

Open gaoyuchris opened this issue 2 years ago • 2 comments

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

gaoyuchris avatar Feb 09 '23 12:02 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.

siyuanliii avatar Feb 09 '23 19:02 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.

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 https://github.com/SysCV/tet/blob/2c18172607d6fc526083c67d93e80104fdc98577/configs/tao/cem_r101_lvis.py#L31). Thus the weights in the repo with the track head part dropped are not the weights I want.

gaoyuchris avatar Feb 10 '23 02:02 gaoyuchris