transfiner icon indicating copy to clipboard operation
transfiner copied to clipboard

Published 20 hours ago • verified publisher icon SysCV

Finetuning on Dataset that only contains human

Open nixczhou opened this issue 3 years ago • 3 comments

Hi, may I know how to finetune MaskTransfiner in dataset that only contains human as the object?

nixczhou avatar Jul 20 '22 04:07 nixczhou

you can set the class number to 1 and delete the original 80 classes embedding weights. Or you set the class index of the human dataset the same as coco index.

lkeab avatar Jul 21 '22 15:07 lkeab

will there be any step by step method to do this? Or do I just have to provide the human dataset in the coco format?

nixczhou avatar Jul 21 '22 16:07 nixczhou

The quickest way is to make you human dataset in coco format. You can refer to more details here.

lkeab avatar Jul 21 '22 16:07 lkeab