我正在使用此代码TensorFlow Feature Extractor使用resnet模型从图像中提取特征CNN特征。这些功能将写入.H5文件。该代码的输入如下
example_feat_extract.py
--network resnet_v1_101
--checkpoint ./checkpoints/resnet_v1_101.ckpt
--image_path ./images_dir/
--out_file ./features.h5
--num_classes 1000
--layer_names resnet_v1_101/logits
当我使用这些输入运行代码时，我得到一个具有nx1项的.H5文件（n是我正在处理的图像总数）。但是，我想要这些图像的实际功能，每个图像的resnet应该为1000。有人可以告诉我输入的“图层名称”应该是什么，以便我得到nx1000大小的特征向量。
谢谢。