Upload Salia_Bam_2.py
Browse files- Salia_Bam_2.py +8 -3
Salia_Bam_2.py
CHANGED
|
@@ -69,7 +69,8 @@ class Salia_BAM_2:
|
|
| 69 |
|
| 70 |
Outputs:
|
| 71 |
Frontview_pos, Sideview_pos, Diagonal_pos, Rearview_pos, image,
|
| 72 |
-
Frontview_neg, Sideview_neg, Diagonal_neg, Rearview_neg, BAM_out
|
|
|
|
| 73 |
|
| 74 |
BAM_in:
|
| 75 |
- If BAM_in == "" (exactly empty), behave normally (use UI inputs).
|
|
@@ -698,12 +699,14 @@ class Salia_BAM_2:
|
|
| 698 |
RETURN_TYPES = (
|
| 699 |
"STRING", "STRING", "STRING", "STRING", "IMAGE",
|
| 700 |
"STRING", "STRING", "STRING", "STRING",
|
| 701 |
-
"STRING" #
|
|
|
|
| 702 |
)
|
| 703 |
RETURN_NAMES = (
|
| 704 |
"Frontview_pos", "Sideview_pos", "Diagonal_pos", "Rearview_pos", "image",
|
| 705 |
"Frontview_neg", "Sideview_neg", "Diagonal_neg", "Rearview_neg",
|
| 706 |
-
"BAM_out"
|
|
|
|
| 707 |
)
|
| 708 |
FUNCTION = "run"
|
| 709 |
|
|
@@ -825,6 +828,7 @@ class Salia_BAM_2:
|
|
| 825 |
# ---------- Bucket selection from age+gender ----------
|
| 826 |
gender_norm = _parse_gender(gender_in)
|
| 827 |
n = _parse_number(age_in)
|
|
|
|
| 828 |
bucket = _bucket_from_number(n)
|
| 829 |
fname = f"{gender_norm}{bucket}.png"
|
| 830 |
|
|
@@ -1091,6 +1095,7 @@ class Salia_BAM_2:
|
|
| 1091 |
Diagonal_neg,
|
| 1092 |
Rearview_neg,
|
| 1093 |
BAM_out,
|
|
|
|
| 1094 |
)
|
| 1095 |
|
| 1096 |
|
|
|
|
| 69 |
|
| 70 |
Outputs:
|
| 71 |
Frontview_pos, Sideview_pos, Diagonal_pos, Rearview_pos, image,
|
| 72 |
+
Frontview_neg, Sideview_neg, Diagonal_neg, Rearview_neg, BAM_out,
|
| 73 |
+
age (INT)
|
| 74 |
|
| 75 |
BAM_in:
|
| 76 |
- If BAM_in == "" (exactly empty), behave normally (use UI inputs).
|
|
|
|
| 699 |
RETURN_TYPES = (
|
| 700 |
"STRING", "STRING", "STRING", "STRING", "IMAGE",
|
| 701 |
"STRING", "STRING", "STRING", "STRING",
|
| 702 |
+
"STRING", # BAM_out
|
| 703 |
+
"INT", # age
|
| 704 |
)
|
| 705 |
RETURN_NAMES = (
|
| 706 |
"Frontview_pos", "Sideview_pos", "Diagonal_pos", "Rearview_pos", "image",
|
| 707 |
"Frontview_neg", "Sideview_neg", "Diagonal_neg", "Rearview_neg",
|
| 708 |
+
"BAM_out",
|
| 709 |
+
"age",
|
| 710 |
)
|
| 711 |
FUNCTION = "run"
|
| 712 |
|
|
|
|
| 828 |
# ---------- Bucket selection from age+gender ----------
|
| 829 |
gender_norm = _parse_gender(gender_in)
|
| 830 |
n = _parse_number(age_in)
|
| 831 |
+
age_out_int = 0 if n is None else int(n)
|
| 832 |
bucket = _bucket_from_number(n)
|
| 833 |
fname = f"{gender_norm}{bucket}.png"
|
| 834 |
|
|
|
|
| 1095 |
Diagonal_neg,
|
| 1096 |
Rearview_neg,
|
| 1097 |
BAM_out,
|
| 1098 |
+
age_out_int,
|
| 1099 |
)
|
| 1100 |
|
| 1101 |
|