We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
Hi,
I'm using the pretrained rtmdet-r_l_syncbn_fast_2xb4-aug-100e_dota model to scan some images and check for just one type of class.
rtmdet-r_l_syncbn_fast_2xb4-aug-100e_dota
I only want to detect one of the many classes it is trained on. How can I do that within my code?
def prepare_inference(): config_file = 'models/rtmdet-r_l_syncbn_fast_2xb4-aug-100e_dota.py' checkpoint_file = 'models/rtmdet-r_l_syncbn_fast_2xb4-aug-100e_dota_20230224_124735-ed4ea966.pth' model = init_detector(config_file, checkpoint_file, device='cuda:0') # or device='cuda:0' return model def start_inference(model, file): result = inference_detector(model, file)
Thanks for any help you can give.
The text was updated successfully, but these errors were encountered:
RangiLyu
No branches or pull requests
Hi,
I'm using the pretrained
rtmdet-r_l_syncbn_fast_2xb4-aug-100e_dota
model to scan some images and check for just one type of class.I only want to detect one of the many classes it is trained on. How can I do that within my code?
Thanks for any help you can give.
The text was updated successfully, but these errors were encountered: