martin/object-segmentation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Adding debug messages

Browse Source
This commit is contained in:
Martin Laasmaa
2025-12-19 10:15:53 +02:00
parent 061f8b3ca2
commit 6ae7481e25
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/model/yolo_wrapper.py
View File

@@ -192,7 +192,9 @@ class YOLOWrapper:
prepared_source, cleanup_path = self._prepare_source(source) prepared_source, cleanup_path = self._prepare_source(source)
try: try:
logger.info(f"Running inference on {source}") logger.info(
f"Running inference on {source} -> prepared_source {prepared_source}"
)
results = self.model.predict( results = self.model.predict(
source=prepared_source, source=prepared_source,
conf=conf, conf=conf,

1
src/utils/ultralytics_16bit_patch.py
View File

@@ -106,6 +106,7 @@ def apply_ultralytics_16bit_tiff_patches(*, force: bool = False) -> None:
def preprocess_batch_16bit(self, batch: dict) -> dict: # type: ignore[override] def preprocess_batch_16bit(self, batch: dict) -> dict: # type: ignore[override]
# Start from upstream behavior to keep device placement + multiscale identical, # Start from upstream behavior to keep device placement + multiscale identical,
# but replace the 255 division with dtype-aware scaling. # but replace the 255 division with dtype-aware scaling.
logger.info(f"Preprocessing batch with monkey-patched preprocess_batch")
for k, v in batch.items(): for k, v in batch.items():
if isinstance(v, torch.Tensor): if isinstance(v, torch.Tensor):
batch[k] = v.to(self.device, non_blocking=self.device.type == "cuda") batch[k] = v.to(self.device, non_blocking=self.device.type == "cuda")