alirezazareian / ovr-cnn

A new framework for open-vocabulary object detection, based on maskrcnn-benchmark
MIT License
226 stars 28 forks source link

Segmentation Mask? #3

Closed lxtGH closed 3 years ago

lxtGH commented 3 years ago

After I pretrain the model using the caption dataset, I train the second stage where I meet such error. Why there are segmentation mask?

image

lxtGH commented 3 years ago

solved by removing mask loading