Naturalize composition frame wording
This commit is contained in:
+4
-1
@@ -145,7 +145,10 @@ def metadata_action_label(row: dict[str, Any], default: str = "sexual pose") ->
|
||||
|
||||
|
||||
def normalize_composition(text: str) -> str:
|
||||
return re.sub(r"^vertical\s+", "", input_policy.clean_text(text), flags=re.IGNORECASE)
|
||||
text = re.sub(r"^vertical\s+", "", input_policy.clean_text(text), flags=re.IGNORECASE)
|
||||
text = re.sub(r"\s+composition$", "", text, flags=re.IGNORECASE)
|
||||
text = re.sub(r"\bcomposition\b", "frame", text, flags=re.IGNORECASE)
|
||||
return text.strip(" ,")
|
||||
|
||||
|
||||
def clean_clothing(text: str) -> str:
|
||||
|
||||
Reference in New Issue
Block a user