小编Rav*_*avi的帖子

TF Slim:在自定义数据集上微调mobilenet v2

我正在尝试在用于图像分类任务的自定义数据集上微调Mobilenet_v2_1.4_224模型。我正在关注本教程TensorFlow-Slim图像分类库。我已经创建了.tfrecord训练和验证文件。当我尝试从现有检查点进行微调时,出现以下错误:

InvalidArgumentError(请参阅上面的回溯):Assign需要两个张量的形状匹配。lhs shape = [1,1,24,144] rhs shape = [1,1,32,192] [[节点:save / Assign_149 =分配[T = DT_FLOAT,_class = [“ loc:@ MobilenetV2 / expanded_conv_2 / expand / weights”] ,use_locking = true,validate_shape = true,_device =“ / job:localhost / replica:0 / task:0 / device:CPU:0”](MobilenetV2 / expanded_conv_2 / expand / weights,保存/恢复V2:149)]]

我使用的微调脚本是:

DATASET_DIR = G:\数据集

TRAIN_DIR = G:\ Dataset \ emotion-models \ mobilenet_v2

CHECKPOINT_PATH = C:\ Users \ lenovo \ Desktop \ mobilenet_v2 \ mobilenet_v2_1.4_224.ckpt

python train_image_classifier.py \
--train_dir=${TRAIN_DIR} \
--dataset_dir=${DATASET_DIR} \ …
Run Code Online (Sandbox Code Playgroud)

python tensorflow tensorflow-slim

2
推荐指数
1
解决办法
3427
查看次数

标签 统计

python ×1

tensorflow ×1

tensorflow-slim ×1