philipp/Abschluss-Projekt
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

training fix.add global settings

Browse Source
This commit is contained in:
Philipp
2025-12-02 09:55:50 +01:00
parent c3c7e042bb
commit 0e31237b79
7 changed files with 96 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
backend/services/generate_yolox_exp.py
View File

@@ -174,10 +174,9 @@ def generate_yolox_inference_exp(training_id, options=None, use_base_config=Fals
'mixup_prob': training.mixup_prob,
'hsv_prob': training.hsv_prob,
'flip_prob': training.flip_prob,
# Convert single values to tuples for YOLOX augmentation parameters
'degrees': (training.degrees, training.degrees) if training.degrees is not None and not isinstance(training.degrees, (list, tuple)) else training.degrees,
'translate': (training.translate, training.translate) if training.translate is not None and not isinstance(training.translate, (list, tuple)) else training.translate,
'shear': (training.shear, training.shear) if training.shear is not None and not isinstance(training.shear, (list, tuple)) else training.shear,
'degrees': training.degrees,
'translate': training.translate,
'shear': training.shear,
'mixup_scale': mixup_scale,
'activation': training.activation,
}
@@ -256,22 +255,11 @@ class Exp(MyExp):
# Format arrays
def format_value(val):
if isinstance(val, (list, tuple)):
# Convert float values to int for size-related parameters
formatted_items = []
for item in val:
# Convert to int if it's a whole number float
if isinstance(item, float) and item.is_integer():
formatted_items.append(str(int(item)))
else:
formatted_items.append(str(item))
return '(' + ', '.join(formatted_items) + ')'
return '(' + ', '.join(map(str, val)) + ')'
elif isinstance(val, bool):
return str(val)
elif isinstance(val, str):
return f'"{val}"'
elif isinstance(val, float) and val.is_integer():
# Convert whole number floats to ints
return str(int(val))
else:
return str(val)