Skip To Content

Точная настройка модели

Вы можете точно настроить модель Prithvi - Flood Segmentation в соответствии с вашим географическим регионом, изображениями и объектами интереса. Точная настройка модели требует меньше обучающих данных, вычислительных ресурсов и времени по сравнению с обучением новой модели.

Точная настройка модели рекомендуется, если вы не получаете удовлетворительных результатов от доступных предварительно обученных моделей глубокого обучения ArcGIS. Это может произойти, если область ваших интересов выходит за пределы географии, применимой к моделям, или если свойства ваших изображений, такие как разрешение, масштаб и сезонность, отличаются.

Вы можете использовать инструмент Экспорт обучающих данных для глубокого обучения для подготовки обучающих данных. Затем вы можете точно настроить эту модель на своих данных с помощью инструмента Тренировать модель глубокого обучения в ArcGIS Pro. Следуйте инструкциям ниже, чтобы точно настроить модель.

Подготовка обучающих данных

Эта модель обучена на 6-канальном составном изображении с Harmonized Landsat 8 (HLSL30) или Harmonized Sentinel 2 (HLSS30) и надписей наводнений. Используйте инструмент Экспорт обучающих данных для глубокого обучения, чтобы подготовить обучающие данные для точной настройки модели.

  1. Перейдите к Инструментам на вкладке Анализ.
    Значок инструментов
  2. Щелкните вкладку Наборы инструментов на панели Геообработка, выберите Инструменты Image Analyst и перейдите к инструменту Экспорт обучающих данных для глубокого обучения в группе инструментов Глубокое обучение.
    Экспорт обучающих данных для глубокого обучения
  3. Установите переменные на вкладке Параметры следующим образом:
    1. Входной растр - выберите 6-канальное изображение. Более подробную информацию о входном растре см. в разделе Рекомендуемая конфигурация изображений.
    2. Выходная папка - любая директория по вашему выбору на вашем компьютере.
    3. Входной класс объектов или Классифицированный растр, или Таблица (дополнительно) - надписанный класс объектов или классифицированный растр, представляющий воду.

      Класс пространственных объектов также должен содержать текстовое поле ClassName со значением Water.

    4. Поле значений класса - это поле ClassValue, задайте значение 1.
    5. Формат изображения - формат TIFF
    6. Размер листа X - 224
    7. Размер листа Y - 224
    8. Шаг по X - 0
    9. Шаг по Y - 0
    10. Формат метаданных - Классифицированные листы
      Параметры инструмента Экспорт обучающих данных для глубокого обучения
  4. Задайте переменные на вкладке Параметры среды:
    1. Экстент обработки — выберите Текущий экстент отображения или любую другую опцию из ниспадающего меню.
    2. Размер ячейки - установите значение желаемого размера ячейки.
      Параметры инструмента Экспорт обучающих данных для глубокого обучения на вкладке Параметры среды
  5. Щелкните Запустить.

    После завершения обработки экспортированные обучающие данные сохраняются в указанной директории.

Точная настройка модели Prithvi - Flood Segmentation

Выполните следующие шаги для точной настройки модели с помощью обучающего модуля ArcGIS API for Python:

  1. Откройте командную строку Python со средой, состоящей из зависимостей глубокого обучения, перейдите в нужную директорию и введите jupyter-notebook.
    Командная строка Python
  2. В браузере щелкните Новый и выберите Python 3 (ipykernel), чтобы создать новый блокнот.
    jupyter notebook
  3. Для точной настройки модели используйте следующие функции:
    1. Импортируйте модель arcgis.learn.
    2. prepare_data - подготовьте объект данных из обучающей выборки, экспортированной с помощью инструмента Экспорт обучающих данных или обучающих выборок в поддерживаемых форматах наборов данных.

      1. path - укажите путь к экспортированным обучающим данным из предыдущего шага.
      2. batch_size - задайте количество листов изображения, обрабатываемых на каждом этапе вывода модели. batch_size зависит от памяти вашей видеокарты.

    3. Инициализируйте модель MMSegmentation и присвойте ей имя переменной, например, model.

      1. data - предоставьте объект данных, созданный с помощью функции prepare_data.
      2. model - задайте имя модели как prithivi100m_crop_classification.

    4. fit - обучите модель для указанного количества периодов времени, используя автоматически сгенерированную оптимальную скорость обучения с помощью метода fit.
    5. save - сохраните обученную модель как пакет глубокого обучения (формат .dlpk), используя метод save. Формат пакета глубокого обучения - это стандартный формат, используемый для развертывания моделей глубокого обучения на платформе ArcGIS. По умолчанию он будет сохранен в подпапке models внутри папки обучающих данных.
    6. per_class_metrics - вычислите точность каждого класса, отзыв и оценку f1 для проверки, заданного с помощью метода per_class_metrics.
      Точная настройка с помощью ArcGIS API for Python

    Теперь вы можете использовать сохраненную модель для анализа ваших изображений.