我正在尝试使用 Faster-RCNN 架构加载检查点并填充模型权重(Faster R-CNN ResNet50 V1 640x640准确地说,从这里开始。我正在尝试加载该网络的权重,类似于RetinaNet 示例笔记本中的操作方式,其中它们请执行下列操作:
fake_box_predictor = tf.compat.v2.train.Checkpoint(
_base_tower_layers_for_heads=detection_model._box_predictor._base_tower_layers_for_heads,
_box_prediction_head=detection_model._box_predictor._box_prediction_head,
)
fake_model = tf.compat.v2.train.Checkpoint(
_feature_extractor=detection_model._feature_extractor,
_box_predictor=fake_box_predictor
)
ckpt = tf.compat.v2.train.Checkpoint(model=fake_model)
ckpt.restore(checkpoint_path).expect_partial()
Run Code Online (Sandbox Code Playgroud)
我正在尝试为我想要使用的 Faster-RCNN 网络获取类似的检查点加载机制,但是 , 等属性_base_tower_layers_for_heads仅存_box_prediction_head在于示例中使用的架构,而不适用于其他任何东西。
我也找不到关于Checkpoint针对我的特定用例填充模型的哪些部分的文档。非常感谢有关如何解决此问题的任何帮助!
我最近刚刚更新了我的 ubuntu 版本,每当我运行任何节点或 npm 命令时,我都会收到此错误。:
node: error while loading shared libraries: libicui18n.so.62: cannot open shared object file: No such file or directory
Run Code Online (Sandbox Code Playgroud)
有什么帮助吗?这似乎是最近的事,我花了很多时间试图弄清楚这一点。