geoai-py 0.5.6__py2.py3-none-any.whl → 0.7.0__py2.py3-none-any.whl

geoai/__init__.py

@@ -2,7 +2,7 @@
 
 __author__ = """Qiusheng Wu"""
 __email__ = "giswqs@gmail.com"
-__version__ = "0.5.6"
+__version__ = "0.7.0"
 
 
 import os

geoai/classify.py

@@ -1,6 +1,7 @@
 """The module for training semantic segmentation models for classifying remote sensing imagery."""
 
 import os
+
 import numpy as np
 
 
@@ -110,23 +111,24 @@ def train_classifier(
     Returns:
         object: Trained SemanticSegmentationTask model.
     """
+    import multiprocessing as mp
+    import timeit
+
+    import albumentations as A
     import lightning.pytorch as pl
+    import torch
+    from lightning.pytorch.callbacks import ModelCheckpoint
+    from lightning.pytorch.loggers import CSVLogger
     from torch.utils.data import DataLoader
-    from torchgeo.datasets import stack_samples, RasterDataset
+    from torchgeo.datamodules import GeoDataModule
+    from torchgeo.datasets import RasterDataset, stack_samples
     from torchgeo.datasets.splits import random_bbox_assignment
     from torchgeo.samplers import (
-        RandomGeoSampler,
-        RandomBatchGeoSampler,
         GridGeoSampler,
+        RandomBatchGeoSampler,
+        RandomGeoSampler,
     )
-    import torch
-    import multiprocessing as mp
-    import timeit
-    import albumentations as A
-    from torchgeo.datamodules import GeoDataModule
     from torchgeo.trainers import SemanticSegmentationTask
-    from lightning.pytorch.callbacks import ModelCheckpoint
-    from lightning.pytorch.loggers import CSVLogger
 
     # Create a wrapper class for albumentations to work with TorchGeo format
     class AlbumentationsWrapper:
@@ -376,21 +378,19 @@ def _classify_image(
     Returns:
         str: Path to the saved classified image.
     """
-    import numpy as np
     import timeit
-    from tqdm import tqdm
 
+    import numpy as np
+    import rasterio
     import torch
+    from rasterio.io import MemoryFile
+    from rasterio.merge import merge
+    from rasterio.transform import from_origin
     from torch.utils.data import DataLoader
-
     from torchgeo.datasets import RasterDataset, stack_samples
     from torchgeo.samplers import GridGeoSampler
     from torchgeo.trainers import SemanticSegmentationTask
-
-    import rasterio
-    from rasterio.transform import from_origin
-    from rasterio.io import MemoryFile
-    from rasterio.merge import merge
+    from tqdm import tqdm
 
     # Set default output path if not provided
     if output_path is None:
@@ -619,14 +619,12 @@ def classify_image(
         str: Path to the saved classified image.
     """
     import timeit
-
-    import torch
-    from torchgeo.trainers import SemanticSegmentationTask
+    import warnings
 
     import rasterio
-
-    import warnings
+    import torch
     from rasterio.errors import NotGeoreferencedWarning
+    from torchgeo.trainers import SemanticSegmentationTask
 
     # Disable specific GDAL/rasterio warnings
     warnings.filterwarnings("ignore", category=UserWarning, module="rasterio._.*")
@@ -863,9 +861,10 @@ def classify_images(
         list: List of paths to the saved classified images.
     """
     # Import required libraries
-    from tqdm import tqdm
     import glob
 
+    from tqdm import tqdm
+
     # Process directory input
     if isinstance(image_paths, str) and os.path.isdir(image_paths):
         # Set default output directory if not provided

geoai/extract.py

@@ -2,6 +2,7 @@
 
 # Standard Library
 import os
+import time
 
 # Third-Party Libraries
 import cv2
@@ -19,7 +20,6 @@ from torchvision.models.detection import (
     maskrcnn_resnet50_fpn,
 )
 from tqdm import tqdm
-import time
 
 # Local Imports
 from .utils import get_raster_stats

geoai/geoai.py

@@ -7,25 +7,31 @@ logging.getLogger("maplibre").setLevel(logging.ERROR)
 import leafmap
 import leafmap.maplibregl as maplibregl
 
+from .classify import classify_image, classify_images, train_classifier
 from .download import (
     download_naip,
     download_overture_buildings,
     download_pc_stac_item,
-    get_overture_data,
     extract_building_stats,
+    get_overture_data,
     pc_collection_list,
     pc_item_asset_list,
-    pc_stac_search,
     pc_stac_download,
+    pc_stac_search,
     read_pc_item_asset,
     view_pc_item,
     view_pc_items,
 )
-from .classify import train_classifier, classify_image, classify_images
 from .extract import *
 from .hf import *
 from .segment import *
-from .train import object_detection, object_detection_batch, train_MaskRCNN_model
+from .train import (
+    object_detection,
+    object_detection_batch,
+    train_MaskRCNN_model,
+    train_segmentation_model,
+    semantic_segmentation,
+)
 from .utils import *