search

microsoft / unilm

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities
https://aka.ms/GeneralAI
MIT License
20.2k stars 2.55k forks source link

Training on custom dataset #372

Closed DaraOrange closed 3 years ago

DaraOrange commented 3 years ago

Hello! Can you explain how to train BeIT on custom dataset? The format is segmentation masks.

donglixp commented 3 years ago

Hi @DaraOrange , the instruction of BEiT pre-training can be found at https://github.com/microsoft/unilm/tree/master/beit#example-pre-training-beit-base-on-imagenet-22k

DaraOrange commented 3 years ago

But it is instruction for image classification, not semantic segmentation