Skip To Content

Ajustar el modelo

Puede ajustar el modelo Prithvi - Segmentación de huellas de incendios para adaptarlo a su área geográfica, imágenes o entidades de interés. El ajuste de un modelo requiere menos datos de entrenamiento, recursos de cómputo y tiempo en comparación con el entrenamiento de un nuevo modelo.

Se recomienda ajustar el modelo si no se obtienen resultados satisfactorios con los modelos de aprendizaje profundo preentrenados de ArcGIS disponibles. Esto puede ocurrir cuando el área de interés queda fuera de las geografías aplicables a los modelos, o si las propiedades de las imágenes, como resolución, escala y estacionalidad, son diferentes.

Puede utilizar la herramienta Exportar datos de entrenamiento para aprendizaje profundo para preparar los datos de entrenamiento. A continuación, puede ajustar este modelo a sus datos mediante la herramienta Entrenar modelo de aprendizaje profundo en ArcGIS Pro. Siga los pasos que se indican a continuación para ajustar el modelo.

Preparar datos de entrenamiento

Este modelo se entrena con seis bandas compuestas de Landsat 8 armonizado (HLSL30) o Sentinel 2 armonizado (HLSS30) y etiquetas de huellas de incendios. Utilice la herramienta Exportar datos de entrenamiento para aprendizaje profundo para preparar los datos de entrenamiento para ajustar el modelo.

  1. Vaya a Herramientas en la pestaña Análisis.
    Icono Herramientas
  2. Haga clic en la pestaña Cajas de herramientas del panel Geoprocesamiento, seleccione Herramientas de Image Analyst y busque la herramienta Exportar datos de entrenamiento para aprendizaje profundo en el conjunto de herramientas Aprendizaje profundo.
    Herramienta Exportar datos de formación para aprendizaje profundo
  3. Establezca las variables de la pestaña Parámetros de la siguiente manera:
    1. Ráster de entrada: seleccione una imagen de 6 bandas. Para obtener más información sobre el ráster de entrada, consulte Configuración de imágenes recomendada.
    2. Carpeta de salida: cualquier directorio de su elección en su máquina.
    3. Clase de entidad de entrada o ráster clasificado o tabla: seleccione la clase de entidad etiquetada o el ráster clasificado que representa las huellas de incendio. En el caso de una clase de entidad, debe incluir un campo de texto denominado ClassName que contiene el nombre y otro campo denominado ClassValue.
    4. Campo de valor de clase: si se utiliza una clase de entidad en el paso anterior, agregue el campo ClassValue que hace referencia al valor de clase asignado a la huella de incendio dentro de la clase de entidad mencionada.
    5. Formato de imagen: formato TIFF
    6. Tamaño de tesela X: 224
    7. Tamaño de tesela Y: 224
    8. Paso X: 0
    9. Paso Y: 0
    10. Formato de metadatos: teselas clasificadas.
      Parámetros de la herramienta Exportar datos de formación para aprendizaje profundo
  4. Establezca las variables de la pestaña Entornos.
    1. Extensión de procesamiento: seleccione Extensión de visualización actual o cualquier otra opción del menú desplegable según sea necesario.
    2. Tamaño de celda: establezca el valor en el tamaño de celda deseado.
      Parámetros de la herramienta Exportar datos de formación para aprendizaje profundo
  5. Haga clic en Ejecutar. Una vez finalizado el procesamiento, los datos de entrenamiento exportados se guardan en el directorio especificado.

Ajustar el modelo Prithvi - Segmentación de huellas de incendios

Complete los siguientes pasos para ajustar el modelo mediante el módulo de aprendizaje de ArcGIS API for Python:

  1. Abra la Ventana de comandos de Python con el entorno que consta de las dependencias de aprendizaje profundo, vaya al directorio deseado y escriba jupyter-notebook.
    Ventana de comandos de Python
  2. En el navegador, haga clic en Nuevo y seleccione Python 3 (ipykernel) para crear un notebook.
    jupyter notebook
  3. Utilice las siguientes funciones para ajustar el modelo:
    1. import arcgis.learn module.
    2. prepare_data: prepare un objeto de datos a partir de la muestra de entrenamiento exportada por la herramienta Exportar datos de entrenamiento para aprendizaje profundo o muestras de entrenamiento en los formatos de datasets admitidos.

      1. path: proporcione la ruta a los datos de entrenamiento exportados en el paso anterior.
      2. batch_size: especifique el número de teselas de imágenes procesadas en cada paso de la inferencia del modelo, y depende de la memoria de su tarjeta gráfica.

    3. Inicialice un modelo MMSegmentation y asígnelo a un nombre de variable, por ejemplo, model

      1. data: proporcione el objeto de datos creado mediante la función prepare_data.
      2. model: especifique el nombre de modelo como prithivi100m_crop_classification.

    4. fit: entrene el modelo para el número especificado de épocas mediante la tasa de aprendizaje óptima generada automáticamente utilizando el método fit.
    5. save: guarde el modelo entrenado como un archivo de paquete de aprendizaje profundo (.dlpk) mediante el método save. El formato de archivo del paquete de aprendizaje profundo es el formato estándar utilizado para implementar modelos de aprendizaje profundo en la plataforma ArcGIS. De forma predeterminada, se guardará en la subcarpeta modelos dentro de la carpeta de datos de entrenamiento.
    6. per_class_metrics: calcule la precisión por clase, recuperación y puntuación f1 en el conjunto de validación mediante el método "per_class_metrics".
      Entrene el modelo mediante ArcGIS API for Python
  4. Ahora puede utilizar el modelo guardado para ejecutar inferencia sobre sus imágenes.