search

Layout-Parser / layout-model-training

The scripts for training Detectron2-based Layout Models on popular layout analysis datasets
198 stars 54 forks source link

custom model #5

Open chris-ml92 opened 2 years ago

chris-ml92 commented 2 years ago

Hi!

I was wondering, is it possible to retrain a model given a custom dataset? I'm trying to use your "annotation-service" for creating and labelling my custom dataset. However I found it difficult to understand if then I'll be able to train the model, since I see that currently only the "PriMa" dataset is supported.

Thanks in advance.

SAIVENKATARAJU commented 2 years ago

Hi @ChristianBCabrera did you able to find workaround for this?. I am also using annotation-service for my custom dataset. are you saying we cannot use annotated for dataset for fine-tuning?.

themantalope commented 2 years ago

Did either of you figure this out? I'm trying to use it for a custom project as well. Seems like it's easy enough to configure Label Studio as needed, but the model training doesn't work, and it's not clear how to trouble shoot it.

I can attach logs from the modeling_container if that helps.