使用MobileNetV2为FasterRcnn提供十分钟培训

2024-04-25 04:58:42 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个应用程序接受过MobileNetV2 using FasterRcnn in Tensorflow.的培训

同样的训练数据和程序,我用MobileNetv1进行训练,效果良好。在

我使用了来自这个page的预训练模型both MobileNetV1 and MobileNetV2.

唯一的问题是,我不能从预先训练的模型的检查点初始化训练MobileNetV2中的所有微调变量。这些变量是MobileNetV2使用FasterRCNN训练的模型所必需的。但它们在预训练模型中不可用(all variables from checkpoint are here

WARNING:root:Variable [MobilenetV2/expanded_conv_15/expand/biases] is not available in checkpoint
WARNING:root:Variable [MobilenetV2/expanded_conv_15/expand/weights] is not available in checkpoint
WARNING:root:Variable [MobilenetV2/expanded_conv_15/project/biases] is not available in checkpoint
WARNING:root:Variable [MobilenetV2/expanded_conv_16/depthwise/biases] is not available in checkpoint
WARNING:root:Variable [MobilenetV2/expanded_conv_16/expand/biases] is not available in checkpoint

我不能只初始化五个变量,但为什么结果如此糟糕。 可能是其他问题吗?在


Tags: in模型isnotrootvariableavailableexpand