tensorflow / tensorflow/models
_feature_extractor from faster-rcnn_resnet is not a trackable object. Impossible to get Weights restored.
Open
Nobody has claimed this yet.
models:research:odapi
type:bug
- Dominant language
- Python
- Stars
- 77.7k
- Forks
- 44.8k
- PR merge metrics
- No merged PRs in 30d
Description
I'm trying to restore the weights via checkpoint. The issue that I come across is that the _feature_extractor is not a trackable object.
tf.train.Checkpoint(_feature_extractor=detection_model._feature_extractor). This just happens with the faster rcnn meta architecture as opposite with the ssd one.
After that, I tried to restore the weights just for the prediction heads (box head and class ones). I'm able to do that but then I have another issue with initializing the variables by passing dummy images.
The environment is the same as the google colab Object Detection API Colab TF2.
Thanks in advance!
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.