Skip to content
Discussion options

You must be logged in to vote
predictor = detection_predictor("db_resnet50", pretrained=False)
weights = torch.load('../weights/db_resnet50-ac60cadc.pt', map_location=device)
predictor.model.load_state_dict(weights)

Replies: 1 comment 2 replies

Comment options

You must be logged in to vote
2 replies
@Rilay9
Comment options

@Rilay9
Comment options

Answer selected by lanesket
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants