Блог

Я хочу обучить свою пользовательскую модель с помощью более быстрой версии RCNN ResNet50 V1, и я выполнил следующую команду:

 Tensorflow/models/research/object_detection/model_main_tf2.py 
   --model_dir=Tensorflow/workspace/models/faster_rcnn_resnet50_v1 
   --pipeline_config_path=Tensorflow/workspace/models/faster_rcnn_resnet50_v1/pipeline.config 
   --num_train_steps=2000
 

Затем я получаю следующую ошибку:

 File "/usr/local/lib/python3.7/dist-packages/object_detection/builders/model_builder.py", line 265, in _check_feature_extractor_exists
    'Tensorflow'.format(feature_extractor_type))
ValueError:  is not supported. See `model_builder.py` 
         for features extractors compatible with different versions of Tensorflow.
 

Кто-нибудь может мне помочь, пожалуйста?

Полученная от вас информация наводит меня на мысль, что в вашем конфигурационном файле неверно указано имя экстрактора функций. Правильным способом было бы указать средство извлечения функций в вашем файле конфигурации следующим образом :

 feature_extractor {
      type: 'faster_rcnn_resnet50_keras'
      batch_norm_trainable: true
    }
 

Установите batch_norm_trainable false значение, если вы выполняете точную настройку для меньшего набора данных.

Пример правильного конфигурационного файла для resnet50_V1 находится здесь.