search

sovit-123 / fasterrcnn-pytorch-training-pipeline

PyTorch Faster R-CNN Object Detection on Custom Dataset
MIT License
208 stars 68 forks source link

Extracting image features #76

Open Codehackerone opened 1 year ago

Codehackerone commented 1 year ago

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

sovit-123 commented 1 year ago

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

Codehackerone commented 1 year ago

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

sovit-123 commented 1 year ago

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

Codehackerone commented 1 year ago

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

sovit-123 commented 1 year ago

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.