Deep Learning - "Unbekannter dataset_type=\"KITTI_rectangles\"

Question

Hallo,

Ich arbeite derzeit an der Schulung eines Deep Learning Modells in ArcGIS Pro, das geothermische Bilder in Landsat 8 Bildern erkennen soll. Ich folge diesem Tutorial: Train Deep Learning Model94Hilfe | Dokumentation .

Ich stecke gerade im letzten Schritt des Prozesses fest, dem Training des Modells selbst. Ich habe einen passenden Ordner für alle exportierten Trainingsdaten erstellt und arbeite jetzt mit Python. Hier ist mein Code unten:

import arcpy

from arcpy.ia import *

# Lizenz der ArcGIS Image Analyst Erweiterung prüfen

arcpy.CheckOutExtension("ImageAnalyst")

# Eingabeparameter definieren

in_folder = "C:\Users\gphy426\Documents\Geothermal Pombo\training_data_geoproc"

out_folder = "C:\Users\gphy426\Documents\Geothermal Pombo\trained_model"

max_epochs = 20

model_type = "FEATURE_CLASSIFIER"

batch_size = 2

arg = "# #"

learning_rate = None

backbone_model = "RESNET34"

pretrained_model = None

validation_percent = 10

stop_training = "STOP_TRAINING"

# Ausführen

TrainDeepLearningModel(in_folder, out_folder, max_epochs, model_type, batch_size, arg, learning_rate, backbone_model, pretrained_model, validation_percent, stop_training) Der Fehler, auf den ich jetzt stoße, lautet "unknown dataset_type = KITTI_rectangles". Ich habe ein Bild des genauen Fehlers, den ich erhalte, an diesen Forenbeitrag angehängt. Ich habe ein wenig im Forum recherchiert, um zu sehen, wie ich dieses Problem umgehen könnte, aber die Hauptlösung, die ich bezüglich pascal VOC Objekten gesehen habe, funktioniert in meinem Fall nicht, da das Referenzraster zwei Bänder hat (nicht optimal sind eins oder drei). Ich stoße auf dasselbe Problem bei der Verwendung des Geoverarbeitungswerkzeugs "train deep learning model". Jegliche Gedanken oder Vorschläge sind willkommen.

SangeetMathew · Answer

Hi Nick,

The training api and tools use  "transfer learning" to train our models which means your data needs to be 3 Band 8 bit. What I would suggest to make your dataset training capable, is to use a composite band tool/function on your dataset to make it 3 band( Here is some help How To: Convert individual raster bands into a color or multiband image ). If it is not 8 bit you can use the stretch function to make it 8 bit. Then run the Export training tool (using PASCAL VOC Objects option ) and then train your data. Hope this helps.

Thanks

Sangeet