FileNotFoundError

[efdalkaya]

Dear Professor Wu,
First of all, thank you so much for your hard work and the amazing contributions you are making to modern geospatial earth!

I am making a tutorial series and watch youtube geoai tutorial series. I run the code vscode and google colab. But i am not running all code. I have a high spatial resolution (0.17 meters) UAV data. TrainData.geojson file including a class (building). I run the following codes in order and finally encountered an error during the execution of the train_MaskRCNN_model() code.

import geoai

train_raster_path = "TrainRasterData.tif"
train_vector_path = "TrainData.geojson"
test_raster_path = "TestRasterData.tif"

out_folder = "output"
tiles = geoai.export_geotiff_tiles(
    in_raster=train_raster_path,
    out_folder=out_folder,
    in_class_data=train_vector_path,
    tile_size=512,
    stride=256,
    buffer_radius=0,
)

geoai.train_MaskRCNN_model(
    images_dir=f"{out_folder}/images",
    labels_dir=f"{out_folder}/labels",
    output_dir=f"{out_folder}/models",
    num_channels=4,
    pretrained=True,
    batch_size=4,
    num_epochs=5,
    learning_rate=0.005,
    val_split=0.2,
)

How can i solve the problem? Thank you.