search

yangxue0827 / RotationDetection

This is a tensorflow-based rotation detection benchmark, also called AlphaRotate.
https://rotationdetection.readthedocs.io/
Apache License 2.0
1.08k stars 182 forks source link

Tensor name "resnet50_v1d/C1/conv0/BatchNorm/beta" not found in checkpoint files /tank/newhome/liuyi/Documents/RotationDetection-main/dataloader/pretrained_weights/resnet50_v1d.ckpt #20

Closed lyccol closed 3 years ago

lyccol commented 3 years ago

tensorflow.python.framework.errors_impl.NotFoundError: Restoring from checkpoint failed. This is most likely due to a Variable name or other graph key that is missing from the checkpoint. Please ensure that you have not altered the graph expected based on the checkpoint. Original error:

Tensor name "resnet50_v1d/C1/conv0/BatchNorm/beta" not found in checkpoint files /tank/newhome/liuyi/Documents/RotationDetection-main/dataloader/pretrained_weights/resnet50_v1d.ckpt [[node save/RestoreV2 (defined at /tank/newhome/liuyi/enter/envs/rod/lib/python3.5/site-packages/tensorflow_core/python/framework/ops.py:1748) ]]

yangxue0827 commented 3 years ago

not found in checkpoint files /tank/newhome/liuyi/Documents/RotationDetection-main/dataloader/pretrained_weights/resnet50_v1d.ckpt

lyccol commented 3 years ago

not found in checkpoint files /tank/newhome/liuyi/Documents/RotationDetection-main/dataloader/pretrained_weights/resnet50_v1d.ckpt

杨博士您好：

Tensorflow pretrain weights: resnet50_v1, resnet101_v1, resnet152_v1, efficientnet, mobilenet_v2, darknet53 (Baidu Drive (1jg2), Google Drive).

我在原文的这个地方点击了resnet50_v1，下载了resnet_v1_50.ckpt。我复制到RotationDetection-main/dataloader/pretrained_weights。改名为resnet50_v1d.ckpt。但是还是显示找不到。

yangxue0827 commented 3 years ago

image resnet50_v1d要从第二个下载地址下载，你要使用resnet_v1_50的话就把cfgs里的resnet50_v1d改成resnet_v1_50就行。但是建议使用resnet50_v1d。