Pikaliov/caption-test
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

checkpoint 15_05

Browse Source
This commit is contained in:
pikaliov
2026-05-15 14:29:42 +03:00
parent d82fac9190
commit ecf8071220
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/datasets/gtauav_dataset.py
View File

@@ -37,6 +37,10 @@ _EMPTY_CAPTION = ""
# Regex to split P1/P2/P3 sections. # Regex to split P1/P2/P3 sections.
_P_SPLIT = re.compile(r"\*\*P[123][^*]*\*\*\s*:?\s*") _P_SPLIT = re.compile(r"\*\*P[123][^*]*\*\*\s*:?\s*")
# TODO: Transforms
# sat_transforms = ...
# drone_transforms = ...
def _parse_caption_levels(output: str) -> tuple[str, str, str]: def _parse_caption_levels(output: str) -> tuple[str, str, str]:
"""Split VLM caption output into L1, L2, L3 levels. """Split VLM caption output into L1, L2, L3 levels.