Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to solve the unregistry bug during training process? #10

Open
kylechang523 opened this issue Jun 29, 2023 · 1 comment
Open

How to solve the unregistry bug during training process? #10

kylechang523 opened this issue Jun 29, 2023 · 1 comment

Comments

@kylechang523
Copy link

Hi,

I want to re-implement the training pipeline and follow the README, but while loading the model I found this bug:

KeyError: "No object named 'SOTR' found in 'META_ARCH' registry!"

May I ask how to solve it?

@nlpkiddo-2001
Copy link

nlpkiddo-2001 commented Sep 10, 2024

+1, Awesome work. Can you share an example script for sample usage. It would be very helpful

`from detectron2.config import get_cfg
from detectron2 import model_zoo
from detectron2.engine import DefaultPredictor

from detectron2.utils.visualizer import Visualizer
from detectron2.data import MetadataCatalog

config = get_cfg()

config.merge_from_file('/DocSegTr/configs/SOTR/R_101_DCN_doc.yaml')
config.MODEL.WEIGHTS = 'document_image_segmentation/model_final_prima_docsegtr.pth'
model = DefaultPredictor(config)
`

and I am getting KeyError: 'Non-existent config key: MODEL.SOTR'

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants