topic Re: Train Deep Learning model with Tree Detection Pre-trained Model in ArcGIS Image Analyst Questions

Train Deep Learning model with Tree Detection Pre-trained Model

SadraDehghan — Tue, 29 Nov 2022 23:42:15 GMT

I'm trying to detect trees for a project from aerial imagery and trying to create a tree detection model that does a close to accurate job (the pretrained tree detection model gets less than half the trees and captures many object that aren't trees). I have set up a small amount of training data (I don't really have time to do much more) and when I train a model with just my training data it does a good job at only identifying trees but misses a lot of them.

Is there a way I could use the pretrained model to finetune the model from the training data I have set up? I tried including the pretrained tree detection model as the advanced option in the "Train Deep Learning Model" tool but I get the following error message:

 File "c:\program files\arcgis\pro\Resources\ArcToolbox\toolboxes\Image Analyst Tools.tbx\TrainDeepLearningModel.tool\tool.script.execute.py", line 327, in execute
    training_model = getattr(arcgis.learn, model_type)
AttributeError: module 'arcgis.learn' has no attribute 'DeepForest'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "c:\program files\arcgis\pro\Resources\ArcToolbox\toolboxes\Image Analyst Tools.tbx\TrainDeepLearningModel.tool\tool.script.execute.py", line 390, in <module>
    execute()
  File "c:\program files\arcgis\pro\Resources\ArcToolbox\toolboxes\Image Analyst Tools.tbx\TrainDeepLearningModel.tool\tool.script.execute.py", line 385, in execute
    del training_model_object
UnboundLocalError: local variable 'training_model_object' referenced before assignment

Thanks!

Re: Train Deep Learning model with Tree Detection Pre-trained Model

PavanYadav — Wed, 22 Mar 2023 16:52:38 GMT

Hello @SadraDehghan. I am assuming you're using the "Pre-trained Model" option available under the advance section. Are you using one of the pre-trained models available on the ArcGIS Living Atlas? Can you please share the name or link to the model? Can you please also share the metadata format used to create your additional training data?

Re: Train Deep Learning model with Tree Detection Pre-trained Model

Student123 — Sun, 09 Jun 2024 11:40:53 GMT

I have the same problem and I dont know how to fix it. I use the Tree Detection Model of Esri (Deep Forest), my ArcGIS Pro Version is 3.2 and I can train my own model from scratch and I can also use this model, but I always get the error when I want to do Transfer Learning with the pretrained Model

Re: Train Deep Learning model with Tree Detection Pre-trained Model

PavanYadav — Mon, 17 Jun 2024 16:30:53 GMT

@Student123 Can you share the .emd file created by the Export Training Data tool?

Re: Train Deep Learning model with Tree Detection Pre-trained Model

VaibhavR — Thu, 20 Jun 2024 06:24:27 GMT

Hi all,

Tree detection model : https://arcg.is/0yLem5
Tree segmentation model : https://arcg.is/1OKPS10
Both are 3rd party models and fine-tuning or transfer learning is not support from them.

Re: Train Deep Learning model with Tree Detection Pre-trained Model

LukasFalk — Mon, 30 Sep 2024 18:47:26 GMT

Pulled from (https://learn.arcgis.com/en/projects/improve-a-deep-learning-model-with-transfer-learning/).

Re: Train Deep Learning model with Tree Detection Pre-trained Model

AndrewBivins — Mon, 07 Oct 2024 20:54:55 GMT

So is there any way to build on these models to advance our tree detection faster? Or do we have to start from scratch?

Thanks,

Re: Train Deep Learning model with Tree Detection Pre-trained Model

PavanYadav — Thu, 31 Oct 2024 21:57:15 GMT

seems like you can finetune the tree detection model - https://doc.arcgis.com/en/pretrained-models/latest/imagery/finetuning-the-palm-tree-detection.htm