fasterrcnn-pytorch-training-pipeline icon indicating copy to clipboard operation
fasterrcnn-pytorch-training-pipeline copied to clipboard

Published 20 hours ago • verified publisher icon sovit-123

Extracting image features

Open Codehackerone opened this issue 2 years ago • 5 comments

Is there a way to extract image features after the fasterrcnn model has been trained?

Codehackerone avatar Jun 20 '23 05:06 Codehackerone

@Codehackerone I have not provided the functionality as of now. But may do it eventually.

sovit-123 avatar Jun 20 '23 06:06 sovit-123

Alright. Is using a hook after the rpn going to work for extracting the features?

Codehackerone avatar Jun 20 '23 06:06 Codehackerone

Yes, if you can modify the code, that can work.

sovit-123 avatar Jun 20 '23 08:06 sovit-123

I have managed to do the feature extraction part... can I contribute here?

Codehackerone avatar Jun 24 '23 07:06 Codehackerone

Sure, if you can create a clean PR, I will be happy to merge it. I made some updates to the training and datasets file. Please make sure to pull the latest code before creating a PR.

sovit-123 avatar Jun 24 '23 08:06 sovit-123