Update predict raster function

.idea/.gitignore

@@ -0,0 +1,3 @@
+# Default ignored files
+/shelf/
+/workspace.xml

.idea/Segment_things.iml

@@ -0,0 +1,14 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$">
+      <excludeFolder url="file://$MODULE_DIR$/venv" />
+    </content>
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+  <component name="PyDocumentationSettings">
+    <option name="format" value="PLAIN" />
+    <option name="myDocStringFormat" value="Plain" />
+  </component>
+</module>

.idea/copilot/chatSessions/xd.lck

@@ -0,0 +1,194 @@
+Private property of Exodus: [email protected]
+
+jetbrains.exodus.io.LockingManager.lock(LockingManager.kt:88)
+jetbrains.exodus.io.LockingManager.lock(LockingManager.kt:39)
+jetbrains.exodus.io.FileDataWriter.lock(FileDataWriter.kt:70)
+jetbrains.exodus.log.Log.tryLock(Log.kt:804)
+jetbrains.exodus.log.Log.<init>(Log.kt:117)
+jetbrains.exodus.env.Environments.newLogInstance(Environments.kt:117)
+jetbrains.exodus.env.Environments.newLogInstance(Environments.kt:81)
+jetbrains.exodus.env.Environments.newLogInstance(Environments.kt:77)
+jetbrains.exodus.env.Environments$newInstance$4.invoke(Environments.kt:46)
+jetbrains.exodus.env.Environments$newInstance$4.invoke(Environments.kt:46)
+jetbrains.exodus.env.Environments.prepare(Environments.kt:120)
+jetbrains.exodus.env.Environments.newInstance(Environments.kt:46)
+kotlinx.dnq.store.container.EntityStoreHelperKt.createTransientEntityStore(EntityStoreHelper.kt:40)
+kotlinx.dnq.store.container.EntityStoreHelperKt.createTransientEntityStore(EntityStoreHelper.kt:31)
+kotlinx.dnq.store.container.EntityStoreHelperKt.createTransientEntityStore$default(EntityStoreHelper.kt:30)
+com.github.copilot.chat.session.persistence.xodus.XdChatSessionPersistenceService.initStore(XdChatSessionPersistenceService.kt:115)
+com.github.copilot.chat.session.persistence.xodus.XdChatSessionPersistenceService.<init>(XdChatSessionPersistenceService.kt:22)
+com.github.copilot.chat.session.persistence.xodus.XdChatSessionPersistenceService.<init>(XdChatSessionPersistenceService.kt:15)
+com.github.copilot.chat.session.persistence.ChatSessionPersistenceServiceKt.ChatSessionPersistenceService(ChatSessionPersistenceService.kt:43)
+com.github.copilot.chat.session.persistence.ChatSessionPersistenceServiceKt.chatSessionsPersistenceService(ChatSessionPersistenceService.kt:53)
+com.github.copilot.chat.session.ChatSessionManager.<init>(ChatSessionManager.kt:45)
+com.github.copilot.chat.session.ChatSessionManager.<init>(ChatSessionManager.kt:25)
+com.github.copilot.chat.window.CopilotChatToolWindow.onCopilotReady(CopilotChatToolWindow.kt:133)
+com.github.copilot.chat.window.CopilotChatToolWindow.access$onCopilotReady(CopilotChatToolWindow.kt:40)
+com.github.copilot.chat.window.CopilotChatToolWindow$initCopilotStatusListener$1.invoke(CopilotChatToolWindow.kt:118)
+com.github.copilot.chat.window.CopilotChatToolWindow$initCopilotStatusListener$1.invoke(CopilotChatToolWindow.kt:115)
+com.github.copilot.status.CopilotAuthStatusKt.subscribeToCopilotAuthStatus(CopilotAuthStatus.kt:27)
+com.github.copilot.chat.window.CopilotChatToolWindow.initCopilotStatusListener(CopilotChatToolWindow.kt:115)
+com.github.copilot.chat.window.CopilotChatToolWindow.<init>(CopilotChatToolWindow.kt:59)
+com.github.copilot.chat.window.CopilotChatToolWindow.<init>(CopilotChatToolWindow.kt:40)
+com.github.copilot.chat.window.CopilotChatToolWindowFactory.init(CopilotChatToolWindowFactory.kt:18)
+com.intellij.openapi.wm.impl.ToolWindowManagerImpl.registerToolWindow$intellij_platform_ide_impl(ToolWindowManagerImpl.kt:1055)
+com.intellij.toolWindow.ToolWindowSetInitializerKt.registerToolWindows(ToolWindowSetInitializer.kt:176)
+com.intellij.toolWindow.ToolWindowSetInitializerKt.access$registerToolWindows(ToolWindowSetInitializer.kt:1)
+com.intellij.toolWindow.ToolWindowSetInitializer$createAndLayoutToolWindows$entries$1.invokeSuspend(ToolWindowSetInitializer.kt:125)
+kotlin.coroutines.jvm.internal.BaseContinuationImpl.resumeWith(ContinuationImpl.kt:33)
+kotlinx.coroutines.DispatchedTask.run(DispatchedTask.kt:106)
+com.intellij.openapi.application.TransactionGuardImpl$1.run(TransactionGuardImpl.java:194)
+com.intellij.openapi.application.impl.ApplicationImpl.runIntendedWriteActionOnCurrentThread(ApplicationImpl.java:831)
+com.intellij.openapi.application.impl.ApplicationImpl$3.run(ApplicationImpl.java:456)
+com.intellij.openapi.application.impl.FlushQueue.doRun(FlushQueue.java:79)
+com.intellij.openapi.application.impl.FlushQueue.runNextEvent(FlushQueue.java:122)
+com.intellij.openapi.application.impl.FlushQueue.flushNow(FlushQueue.java:41)
+java.desktop/java.awt.event.InvocationEvent.dispatch(InvocationEvent.java:318)
+java.desktop/java.awt.EventQueue.dispatchEventImpl(EventQueue.java:788)
+java.desktop/java.awt.EventQueue$3.run(EventQueue.java:739)
+java.desktop/java.awt.EventQueue$3.run(EventQueue.java:731)
+java.base/java.security.AccessController.doPrivileged(AccessController.java:399)
+java.base/java.security.ProtectionDomain$JavaSecurityAccessImpl.doIntersectionPrivilege(ProtectionDomain.java:86)
+java.desktop/java.awt.EventQueue.dispatchEvent(EventQueue.java:758)
+com.intellij.ide.IdeEventQueue.defaultDispatchEvent(IdeEventQueue.kt:666)
+com.intellij.ide.IdeEventQueue._dispatchEvent$lambda$7(IdeEventQueue.kt:570)
+com.intellij.openapi.application.impl.ApplicationImpl.withoutImplicitRead(ApplicationImpl.java:1446)
+com.intellij.ide.IdeEventQueue._dispatchEvent(IdeEventQueue.kt:570)
+com.intellij.ide.IdeEventQueue.access$_dispatchEvent(IdeEventQueue.kt:68)
+com.intellij.ide.IdeEventQueue$dispatchEvent$processEventRunnable$1$1$1.compute(IdeEventQueue.kt:349)
+com.intellij.ide.IdeEventQueue$dispatchEvent$processEventRunnable$1$1$1.compute(IdeEventQueue.kt:348)
+com.intellij.openapi.progress.impl.CoreProgressManager.computePrioritized(CoreProgressManager.java:787)
+com.intellij.ide.IdeEventQueue$dispatchEvent$processEventRunnable$1$1.invoke(IdeEventQueue.kt:348)
+com.intellij.ide.IdeEventQueue$dispatchEvent$processEventRunnable$1$1.invoke(IdeEventQueue.kt:343)
+com.intellij.ide.IdeEventQueueKt.performActivity$lambda$1(IdeEventQueue.kt:994)
+com.intellij.openapi.application.TransactionGuardImpl.performActivity(TransactionGuardImpl.java:113)
+com.intellij.ide.IdeEventQueueKt.performActivity(IdeEventQueue.kt:994)
+com.intellij.ide.IdeEventQueue.dispatchEvent$lambda$4(IdeEventQueue.kt:343)
+com.intellij.openapi.application.impl.ApplicationImpl.runIntendedWriteActionOnCurrentThread(ApplicationImpl.java:831)
+com.intellij.ide.IdeEventQueue.dispatchEvent(IdeEventQueue.kt:385)
+com.intellij.openapi.progress.impl.PlatformTaskSupportKt.pumpEventsForHierarchy(PlatformTaskSupport.kt:341)
+com.intellij.openapi.progress.impl.PlatformTaskSupportKt.access$pumpEventsForHierarchy(PlatformTaskSupport.kt:1)
+com.intellij.openapi.progress.impl.PlatformTaskSupport$runBlockingModalInternal$2$1.invoke(PlatformTaskSupport.kt:139)
+com.intellij.openapi.progress.impl.PlatformTaskSupport$runBlockingModalInternal$2$1.invoke(PlatformTaskSupport.kt:121)
+com.intellij.openapi.application.impl.ModalityKt.inModalContext(modality.kt:36)
+com.intellij.openapi.progress.impl.PlatformTaskSupport.runBlockingModalInternal(PlatformTaskSupport.kt:121)
+com.intellij.openapi.progress.impl.PlatformTaskSupport.access$runBlockingModalInternal(PlatformTaskSupport.kt:41)
+com.intellij.openapi.progress.impl.PlatformTaskSupport$runBlockingModalInternal$1.invoke(PlatformTaskSupport.kt:113)
+com.intellij.openapi.progress.impl.PlatformTaskSupport$runBlockingModalInternal$1.invoke(PlatformTaskSupport.kt:110)
+com.intellij.openapi.progress.CancellationKt$ensureCurrentJob$1.invoke(cancellation.kt:68)
+com.intellij.openapi.progress.CancellationKt.withCurrentJob$lambda$0(cancellation.kt:17)
+com.intellij.openapi.progress.Cancellation.withCurrentJob(Cancellation.java:60)
+com.intellij.openapi.progress.CancellationKt.withCurrentJob(cancellation.kt:17)
+com.intellij.openapi.progress.CancellationKt.executeWithJobAndCompleteIt(cancellation.kt:126)
+com.intellij.openapi.progress.CancellationKt.ensureCurrentJob(cancellation.kt:67)
+com.intellij.openapi.progress.CancellationKt.ensureCurrentJobAllowingOrphan(cancellation.kt:51)
+com.intellij.openapi.progress.impl.PlatformTaskSupport.runBlockingModalInternal(PlatformTaskSupport.kt:110)
+com.intellij.openapi.progress.TasksKt.runBlockingModalWithRawProgressReporter(tasks.kt:263)
+com.intellij.ide.impl.ProjectUtilKt.runBlockingUnderModalProgress(ProjectUtil.kt:731)
+com.intellij.ide.impl.ProjectUtilKt.runBlockingUnderModalProgress$default(ProjectUtil.kt:726)
+com.intellij.ide.impl.ProjectUtilKt.runUnderModalProgressIfIsEdt(ProjectUtil.kt:718)
+com.intellij.openapi.project.impl.ProjectManagerImpl.openProject(ProjectManagerImpl.kt:521)
+com.intellij.platform.PlatformProjectOpenProcessor$Companion.doOpenProject(PlatformProjectOpenProcessor.kt:173)
+com.intellij.ide.util.projectWizard.AbstractNewProjectStep.doGenerateProject(AbstractNewProjectStep.java:241)
+com.intellij.ide.util.projectWizard.AbstractNewProjectStep.doGenerateProject(AbstractNewProjectStep.java:189)
+com.jetbrains.python.newProject.steps.PythonGenerateProjectCallback.generateProject(PythonGenerateProjectCallback.java:78)
+com.jetbrains.python.newProject.steps.PythonGenerateProjectCallback.consume(PythonGenerateProjectCallback.java:56)
+com.intellij.ide.util.projectWizard.ProjectSettingsStepBase$1.actionPerformed(ProjectSettingsStepBase.java:148)
+java.desktop/javax.swing.AbstractButton.fireActionPerformed(AbstractButton.java:1972)
+java.desktop/javax.swing.AbstractButton$Handler.actionPerformed(AbstractButton.java:2313)
+java.desktop/javax.swing.DefaultButtonModel.fireActionPerformed(DefaultButtonModel.java:405)
+java.desktop/javax.swing.DefaultButtonModel.setPressed(DefaultButtonModel.java:262)
+java.desktop/javax.swing.plaf.basic.BasicButtonListener.mouseReleased(BasicButtonListener.java:279)
+java.desktop/java.awt.Component.processMouseEvent(Component.java:6656)
+java.desktop/javax.swing.JComponent.processMouseEvent(JComponent.java:3385)
+java.desktop/java.awt.Component.processEvent(Component.java:6421)
+java.desktop/java.awt.Container.processEvent(Container.java:2266)
+java.desktop/java.awt.Component.dispatchEventImpl(Component.java:5026)
+java.desktop/java.awt.Container.dispatchEventImpl(Container.java:2324)
+java.desktop/java.awt.Component.dispatchEvent(Component.java:4854)
+java.desktop/java.awt.LightweightDispatcher.retargetMouseEvent(Container.java:4948)
+java.desktop/java.awt.LightweightDispatcher.processMouseEvent(Container.java:4575)
+java.desktop/java.awt.LightweightDispatcher.dispatchEvent(Container.java:4516)
+java.desktop/java.awt.Container.dispatchEventImpl(Container.java:2310)
+java.desktop/java.awt.Window.dispatchEventImpl(Window.java:2804)
+java.desktop/java.awt.Component.dispatchEvent(Component.java:4854)
+java.desktop/java.awt.EventQueue.dispatchEventImpl(EventQueue.java:790)
+java.desktop/java.awt.EventQueue$3.run(EventQueue.java:739)
+java.desktop/java.awt.EventQueue$3.run(EventQueue.java:731)
+java.base/java.security.AccessController.doPrivileged(AccessController.java:399)
+java.base/java.security.ProtectionDomain$JavaSecurityAccessImpl.doIntersectionPrivilege(ProtectionDomain.java:86)
+java.base/java.security.ProtectionDomain$JavaSecurityAccessImpl.doIntersectionPrivilege(ProtectionDomain.java:97)
+java.desktop/java.awt.EventQueue$4.run(EventQueue.java:763)
+java.desktop/java.awt.EventQueue$4.run(EventQueue.java:761)
+java.base/java.security.AccessController.doPrivileged(AccessController.java:399)
+java.base/java.security.ProtectionDomain$JavaSecurityAccessImpl.doIntersectionPrivilege(ProtectionDomain.java:86)
+java.desktop/java.awt.EventQueue.dispatchEvent(EventQueue.java:760)
+com.intellij.ide.IdeEventQueue.defaultDispatchEvent(IdeEventQueue.kt:666)
+com.intellij.ide.IdeEventQueue.dispatchMouseEvent(IdeEventQueue.kt:614)
+com.intellij.ide.IdeEventQueue._dispatchEvent(IdeEventQueue.kt:569)
+com.intellij.ide.IdeEventQueue.access$_dispatchEvent(IdeEventQueue.kt:68)
+com.intellij.ide.IdeEventQueue$dispatchEvent$processEventRunnable$1$1$1.compute(IdeEventQueue.kt:349)
+com.intellij.ide.IdeEventQueue$dispatchEvent$processEventRunnable$1$1$1.compute(IdeEventQueue.kt:348)
+com.intellij.openapi.progress.impl.CoreProgressManager.computePrioritized(CoreProgressManager.java:787)
+com.intellij.ide.IdeEventQueue$dispatchEvent$processEventRunnable$1$1.invoke(IdeEventQueue.kt:348)
+com.intellij.ide.IdeEventQueue$dispatchEvent$processEventRunnable$1$1.invoke(IdeEventQueue.kt:343)
+com.intellij.ide.IdeEventQueueKt.performActivity$lambda$1(IdeEventQueue.kt:994)
+com.intellij.openapi.application.TransactionGuardImpl.performActivity(TransactionGuardImpl.java:105)
+com.intellij.ide.IdeEventQueueKt.performActivity(IdeEventQueue.kt:994)
+com.intellij.ide.IdeEventQueue.dispatchEvent$lambda$4(IdeEventQueue.kt:343)
+com.intellij.openapi.application.impl.ApplicationImpl.runIntendedWriteActionOnCurrentThread(ApplicationImpl.java:831)
+com.intellij.ide.IdeEventQueue.dispatchEvent(IdeEventQueue.kt:385)
+java.desktop/java.awt.EventDispatchThread.pumpOneEventForFilters(EventDispatchThread.java:207)
+java.desktop/java.awt.EventDispatchThread.pumpEventsForFilter(EventDispatchThread.java:128)
+java.desktop/java.awt.EventDispatchThread.pumpEventsForFilter(EventDispatchThread.java:121)
+java.desktop/java.awt.WaitDispatchSupport$2.run(WaitDispatchSupport.java:191)
+java.desktop/java.awt.WaitDispatchSupport$4.run(WaitDispatchSupport.java:236)
+java.desktop/java.awt.WaitDispatchSupport$4.run(WaitDispatchSupport.java:234)
+java.base/java.security.AccessController.doPrivileged(AccessController.java:318)
+java.desktop/java.awt.WaitDispatchSupport.enter(WaitDispatchSupport.java:234)
+java.desktop/java.awt.Dialog.show(Dialog.java:1080)
+com.intellij.openapi.ui.impl.DialogWrapperPeerImpl$MyDialog.show(DialogWrapperPeerImpl.java:741)
+com.intellij.openapi.ui.impl.DialogWrapperPeerImpl.show(DialogWrapperPeerImpl.java:466)
+com.intellij.openapi.ui.DialogWrapper.doShow(DialogWrapper.java:1675)
+com.intellij.openapi.ui.DialogWrapper.show(DialogWrapper.java:1633)
+com.jetbrains.python.newProject.PyCharmNewProjectAction.actionPerformed(PyCharmNewProjectAction.java:13)
+com.intellij.openapi.actionSystem.ex.ActionUtil.doPerformActionOrShowPopup(ActionUtil.java:333)
+com.intellij.openapi.actionSystem.ex.ActionUtil.lambda$performActionDumbAwareWithCallbacks$4(ActionUtil.java:307)
+com.intellij.openapi.actionSystem.ex.ActionUtil.performDumbAwareWithCallbacks(ActionUtil.java:356)
+com.intellij.openapi.actionSystem.ex.ActionUtil.performActionDumbAwareWithCallbacks(ActionUtil.java:307)
+com.intellij.openapi.actionSystem.impl.ActionMenuItem.lambda$performAction$5(ActionMenuItem.java:299)
+com.intellij.openapi.wm.impl.FocusManagerImpl.runOnOwnContext(FocusManagerImpl.java:225)
+com.intellij.openapi.actionSystem.impl.ActionMenuItem.performAction(ActionMenuItem.java:292)
+com.intellij.openapi.actionSystem.impl.ActionMenuItem.lambda$new$1(ActionMenuItem.java:81)
+com.intellij.openapi.application.TransactionGuardImpl.performActivity(TransactionGuardImpl.java:113)
+com.intellij.openapi.application.TransactionGuardImpl.performUserActivity(TransactionGuardImpl.java:94)
+com.intellij.openapi.actionSystem.impl.ActionMenuItem.lambda$new$2(ActionMenuItem.java:81)
+java.desktop/java.awt.event.InvocationEvent.dispatch(InvocationEvent.java:318)
+java.desktop/java.awt.EventQueue.dispatchEventImpl(EventQueue.java:788)
+java.desktop/java.awt.EventQueue$3.run(EventQueue.java:739)
+java.desktop/java.awt.EventQueue$3.run(EventQueue.java:731)
+java.base/java.security.AccessController.doPrivileged(AccessController.java:399)
+java.base/java.security.ProtectionDomain$JavaSecurityAccessImpl.doIntersectionPrivilege(ProtectionDomain.java:86)
+java.desktop/java.awt.EventQueue.dispatchEvent(EventQueue.java:758)
+com.intellij.ide.IdeEventQueue.defaultDispatchEvent(IdeEventQueue.kt:666)
+com.intellij.ide.IdeEventQueue._dispatchEvent$lambda$7(IdeEventQueue.kt:570)
+com.intellij.openapi.application.impl.ApplicationImpl.withoutImplicitRead(ApplicationImpl.java:1446)
+com.intellij.ide.IdeEventQueue._dispatchEvent(IdeEventQueue.kt:570)
+com.intellij.ide.IdeEventQueue.access$_dispatchEvent(IdeEventQueue.kt:68)
+com.intellij.ide.IdeEventQueue$dispatchEvent$processEventRunnable$1$1$1.compute(IdeEventQueue.kt:349)
+com.intellij.ide.IdeEventQueue$dispatchEvent$processEventRunnable$1$1$1.compute(IdeEventQueue.kt:348)
+com.intellij.openapi.progress.impl.CoreProgressManager.computePrioritized(CoreProgressManager.java:787)
+com.intellij.ide.IdeEventQueue$dispatchEvent$processEventRunnable$1$1.invoke(IdeEventQueue.kt:348)
+com.intellij.ide.IdeEventQueue$dispatchEvent$processEventRunnable$1$1.invoke(IdeEventQueue.kt:343)
+com.intellij.ide.IdeEventQueueKt.performActivity$lambda$1(IdeEventQueue.kt:994)
+com.intellij.openapi.application.TransactionGuardImpl.performActivity(TransactionGuardImpl.java:105)
+com.intellij.ide.IdeEventQueueKt.performActivity(IdeEventQueue.kt:994)
+com.intellij.ide.IdeEventQueue.dispatchEvent$lambda$4(IdeEventQueue.kt:343)
+com.intellij.openapi.application.impl.ApplicationImpl.runIntendedWriteActionOnCurrentThread(ApplicationImpl.java:831)
+com.intellij.ide.IdeEventQueue.dispatchEvent(IdeEventQueue.kt:385)
+java.desktop/java.awt.EventDispatchThread.pumpOneEventForFilters(EventDispatchThread.java:207)
+java.desktop/java.awt.EventDispatchThread.pumpEventsForFilter(EventDispatchThread.java:128)
+java.desktop/java.awt.EventDispatchThread.pumpEventsForHierarchy(EventDispatchThread.java:117)
+java.desktop/java.awt.EventDispatchThread.pumpEvents(EventDispatchThread.java:113)
+java.desktop/java.awt.EventDispatchThread.pumpEvents(EventDispatchThread.java:105)
+java.desktop/java.awt.EventDispatchThread.run(EventDispatchThread.java:92)

.idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,24 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredPackages">
+        <value>
+          <list size="11">
+            <item index="0" class="java.lang.String" itemvalue="Flask-Caching" />
+            <item index="1" class="java.lang.String" itemvalue="fiona" />
+            <item index="2" class="java.lang.String" itemvalue="geopandas" />
+            <item index="3" class="java.lang.String" itemvalue="basemap" />
+            <item index="4" class="java.lang.String" itemvalue="gdal" />
+            <item index="5" class="java.lang.String" itemvalue="datashader" />
+            <item index="6" class="java.lang.String" itemvalue="statsmodels" />
+            <item index="7" class="java.lang.String" itemvalue="dask" />
+            <item index="8" class="java.lang.String" itemvalue="dash" />
+            <item index="9" class="java.lang.String" itemvalue="dash-bootstrap-components" />
+            <item index="10" class="java.lang.String" itemvalue="imbalanced-learn" />
+          </list>
+        </value>
+      </option>
+    </inspection_tool>
+  </profile>
+</component>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/misc.xml

@@ -0,0 +1,4 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.10 (Segment_things)" project-jdk-type="Python SDK" />
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/Segment_things.iml" filepath="$PROJECT_DIR$/.idea/Segment_things.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+  </component>
+</project>

semantic-segmentation/SemanticModel/prediction.py

@@ -112,91 +112,195 @@ class PredictionPipeline:
         print(f'\nPredictions saved to: {output_dir}')
         return output_dir
 
-    def predict_raster(self, raster_path, tile_size=1024, overlap=0.175,
-                      boundary_path=None, output_path=None, format='integer'):
-        """Processes large raster images using tiling approach."""
-        print('Loading raster...')
+    def predict_raster(
+            self,
+            raster_path,
+            tile_size=1024,
+            overlap=0.175,
+            boundary_path=None,
+            output_path=None,
+            format='integer'
+    ):
+        """
+        Processes large raster images using a tiling approach. For each tile:
+          1) Optionally checks a boundary mask (if provided) to skip tiles outside an ROI.
+          2) Applies augmentations/preprocessing, then runs the model prediction.
+          3) Resizes back to the tile's original size if necessary (e.g., after aug).
+          4) Merges the tile predictions into a final 'pred_raster' (with confidence blending).
+
+        Args:
+            raster_path (str): Path to the large raster image (GeoTIFF).
+            tile_size (int): Dimensions of each tile (default 1024).
+            overlap (float): Overlap fraction between tiles (default 0.175).
+            boundary_path (str): Path to shapefile/geojson for boundary region (optional).
+            output_path (str): Path to save prediction (optional).
+            format (str): 'integer' for integer mask, 'color' for RGB, etc.
+
+        Returns:
+            pred_raster (np.ndarray): 2D or 3D numpy array with the final merged prediction.
+            profile (dict): Raster profile/metadata for georeferencing or saving.
+        """
+
+        print("Loading raster...")
         with rio.open(raster_path) as src:
+            # Read [Bands, Height, Width] -> Move axis => [Height, Width, Bands]
             raster = src.read()
-            raster = np.moveaxis(raster, 0, 2)[:,:,:3]
+            raster = np.moveaxis(raster, 0, 2)
+            raster = raster[:, :, :3]  # keep only first 3 bands if >3
             profile = src.profile
             transform = src.transform
-        
+
+        boundary_geom = None
         if boundary_path:
             boundary = gpd.read_file(boundary_path)
             boundary = boundary.to_crs(profile['crs'])
             boundary_geom = boundary.iloc[0].geometry
-        
+
+        print("Generating tiles...")
         tiles = slidingwindow.generate(
             raster,
             slidingwindow.DimOrder.HeightWidthChannel,
             tile_size,
             overlap
         )
-        
-        pred_raster = np.zeros_like(raster[:,:,0], dtype='uint8')
-        confidence = np.zeros_like(pred_raster, dtype=np.float32)
-        
+
+        # Prepare the output arrays:
+        # pred_raster => final integer (or color) predictions
+        # confidence => track confidence per pixel to do max merging
+        pred_raster = np.zeros_like(raster[:, :, 0], dtype='uint8')  # shape: (H, W)
+        confidence = np.zeros_like(pred_raster, dtype=np.float32)  # shape: (H, W)
+
+        # Get your augmentations/preprocessing
         aug = get_validation_augmentations(tile_size, tile_size, fixed_size=False)
-        
+
+        # -------------------------------
+        # Iterate over each tile
+        # -------------------------------
         for idx, tile in enumerate(tiles):
             if (idx + 1) % 10 == 0 or idx == len(tiles) - 1:
-                print(f'\rProcessed {idx+1}/{len(tiles)} tiles', end='')
-            
+                print(f"\rProcessed {idx + 1}/{len(tiles)} tiles", end="")
+
+            # tile.indices() = (slice(row_start, row_end), slice(col_start, col_end))
             bounds = tile.indices()
-            
+
+            # Extract tile from the big raster
             tile_image = raster[bounds[0], bounds[1]]
-            
-            if boundary_path:
+            # tile_image.shape => (tile_height, tile_width, channels)
+
+            # print(f"\n[DEBUG] Tile #{idx}: tile_image shape before aug = {tile_image.shape}")
+
+            # 1) Check for zero-size tile BEFORE augmentations
+            if tile_image.shape[0] == 0 or tile_image.shape[1] == 0:
+                # print("[DEBUG] Skipping tile => zero dimension BEFORE aug")
+                continue
+
+            # 2) If boundary is given, skip tile if it doesn't intersect
+            if boundary_geom is not None:
                 corners = [
                     convert_coordinates(transform, bounds[1].start, bounds[0].start),
                     convert_coordinates(transform, bounds[1].stop, bounds[0].start),
                     convert_coordinates(transform, bounds[1].stop, bounds[0].stop),
                     convert_coordinates(transform, bounds[1].start, bounds[0].stop)
                 ]
-                if not Polygon(corners).intersects(boundary_geom):
+                poly = Polygon(corners)
+                if not poly.intersects(boundary_geom):
+                    # print("[DEBUG] Skipping tile => outside boundary")
                     continue
-            
+
+            # 3) Apply augmentations
             processed = aug(image=tile_image)['image']
+            # print(f"[DEBUG] processed shape after aug = {processed.shape}")
+
+            # Check for zero-size tile AFTER augmentations
+            if processed.shape[0] == 0 or processed.shape[1] == 0:
+                # print("[DEBUG] Skipping tile => zero dimension AFTER aug")
+                continue
+
+            # 4) Preprocessing for the model
             processed = self.config.preprocessing(image=processed)['image']
-            
             x_tensor = torch.from_numpy(processed).to(self.device).unsqueeze(0)
+
+            # right after model inference and before merging into pred_raster ...
+
             with torch.no_grad():
+                # Model output: shape ~ (1, n_classes, H_aug, W_aug)
                 prediction = self.model.predict(x_tensor)
+                # Remove batch dimension: (n_classes, H_aug, W_aug) for multi-class
                 prediction = prediction.squeeze().cpu().numpy()
-            
-            if self.config.n_classes > 1:
-                tile_pred = np.argmax(prediction, axis=0)
-                tile_conf = np.max(prediction, axis=0)
+
+            # -----------------------------------------------------------------
+            # 1) Convert raw logits -> label map (tile_pred) and confidence map
+            # -----------------------------------------------------------------
+            # If you have 'n_classes=4', `prediction.shape` might be (4, H_aug, W_aug).
+            # We must ARGMAX across the class dimension to get a 2D label map.
+            # In predict_raster() function, replace this part:
+
+            if prediction.ndim == 3 and prediction.shape[0] == self.config.n_classes:
+                # Multi-class case
+                tile_pred = np.argmax(prediction, axis=0).astype(np.uint8)
+                tile_conf = np.max(prediction, axis=0).astype(np.float32)
             else:
-                tile_conf = np.abs(prediction - 0.5)
-                tile_pred = prediction.round()
-            
-            if tile_pred.shape != tile_image.shape[:2]:
-                tile_pred = cv2.resize(tile_pred, tile_image.shape[:2][::-1],
-                                     interpolation=cv2.INTER_NEAREST)
-                tile_conf = cv2.resize(tile_conf, tile_image.shape[:2][::-1],
-                                     interpolation=cv2.INTER_LINEAR)
-            
-            # Update prediction and confidence maps
+                # Binary case - take first channel if multiple channels
+                if prediction.ndim == 3:
+                    prediction = prediction[0]  # Take first channel
+                tile_conf = np.abs(prediction - 0.5).astype(np.float32)
+                tile_pred = np.round(prediction).astype(np.uint8)
+
+            orig_hw = tile_image.shape[:2]  # (tile_height, tile_width)
+            if tile_pred.shape != orig_hw:
+                # print(f"[DEBUG] Resizing from {tile_pred.shape} to {orig_hw} ...")
+
+                # Cast to float32 for cv2.resize
+                tile_pred_float = tile_pred.astype(np.float32)
+                tile_conf_float = tile_conf.astype(np.float32)
+
+                # cv2 expects (width, height)
+                cv2_size = (orig_hw[1], orig_hw[0])
+                if cv2_size[0] == 0 or cv2_size[1] == 0:
+                    # print("[DEBUG] Skipping tile => zero dimension for cv2_size!")
+                    continue
+
+                # NEAREST for label map, LINEAR for confidence
+                tile_pred_resized = cv2.resize(
+                    tile_pred_float, cv2_size, interpolation=cv2.INTER_NEAREST
+                )
+                tile_conf_resized = cv2.resize(
+                    tile_conf_float, cv2_size, interpolation=cv2.INTER_LINEAR
+                )
+
+                # Convert back to appropriate dtypes
+                tile_pred = np.round(tile_pred_resized).astype(np.uint8)
+                tile_conf = tile_conf_resized.astype(np.float32)
+
+            # -----------------------------------------------------------------
+            # 3) Merge tile_pred into the final pred_raster / confidence arrays
+            # -----------------------------------------------------------------
             existing_conf = confidence[bounds[0], bounds[1]]
             existing_pred = pred_raster[bounds[0], bounds[1]]
-            
-            mask = existing_conf < tile_conf
+
+            mask = tile_conf > existing_conf
             existing_pred[mask] = tile_pred[mask]
             existing_conf[mask] = tile_conf[mask]
-            
+
             pred_raster[bounds[0], bounds[1]] = existing_pred
             confidence[bounds[0], bounds[1]] = existing_conf
-        
+
+
+        print("\n Finished all tiles")
+
+        # 9) Convert pred_raster to final format (color or integer)
         pred_raster = self._format_prediction(pred_raster, format)
-        
+
+        # 10) (Optional) Save if output_path or boundary_path provided
         if output_path or boundary_path:
             self._save_raster_prediction(
-                pred_raster, raster_path, output_path,
-                profile, boundary_geom if boundary_path else None
+                pred_raster,
+                raster_path,
+                output_path,
+                profile,
+                boundary_geom if boundary_path else None
             )
-        
+
         return pred_raster, profile
 
     def _format_prediction(self, prediction, format):