Fehler beim Laden von finetune_checkpoint im Objekterkennungstraining

Question

Ich folge diesem tutorial.

Ich verwende Oxford-IIIT Pet Daten und Konfigurationsdatei ähnlich wie this.

In der pipeline_config Datei habe ich Feinabstimmung Kontrollpunkt als

fine_tune_checkpoint: "{PATH TO}/ssd_mobilenet_v1_coco_11_06_2017/model.ckpt" 
from_detection_checkpoint: false 

angegeben Aber wenn ich laufe, Python object_detection/train.py ... --train_dir = $ {} PATH_TO_TRAIN_DIR

Ich erhalte eine Liste mit Warnungen.

WARNUNG: root: Variable [MobalenetV1/Conv2d_0/BatchNorm/beta] ist im Prüfpunkt nicht verfügbar.

.

.

.

ACHTUNG: root: Variable [MobilenetV1/Conv2d_9_pointwise/Gewichte] in Kontrollpunkt nicht verfügbar

und Programm beendet sagen

Valueerror: Keine Variablen

speichern Ich habe überprüft die ckpt in python mit

print_tensors_in_checkpoint_file('{PATH TO}/ssd_mobilenet_v1_coco_11_06_2017/model.ckpt/model.ckpt', '',"") 

Die Ausgabe war eine Liste.

FeatureExtractor/MobilenetV1/Conv2d_0/BatchNorm/beta (DT_FLOAT) [32]

.

.

.

FeatureExtractor/MobilenetV1/Conv2d_9_pointwise/Gewichte (DT_FLOAT) [1,1,512,512]

Hinweis: Training Skript läuft gut, wenn finetune_checkpoint von

from_detection_checkpoint: true 

in Pipeline-Konfigurationsdatei deaktiviert

Answer 1

Die Einstellung from_detection_checkpoint: true ist für die Prüfpunktdatei erforderlich, die Sie in diesem Modus verwenden Ich wurde als Detektor ausgebildet (also wird nichts deaktiviert).

Sie würden nur from_detection_checkpoint: false einstellen, wenn Sie von einem der Slim models, die für ImageNet-Klassifizierung vortrainiert wurde, trainieren wollten.