Commit
·
2403575
1
Parent(s):
da5c717
tested replaycompose
Browse files- src/app.py +9 -1
src/app.py
CHANGED
|
@@ -37,7 +37,15 @@ param_values = show_transform_control(augmentations[transform_name])
|
|
| 37 |
|
| 38 |
# apply the transformation to the image
|
| 39 |
transform = getattr(A, transform_name)(**param_values)
|
| 40 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
|
| 42 |
|
| 43 |
# show the images
|
|
|
|
| 37 |
|
| 38 |
# apply the transformation to the image
|
| 39 |
transform = getattr(A, transform_name)(**param_values)
|
| 40 |
+
data = A.ReplayCompose([transform])(image=image)
|
| 41 |
+
augmented_image = data["image"]
|
| 42 |
+
|
| 43 |
+
# TODO add convinient replay compose
|
| 44 |
+
# applied_params = data["replay"]["transforms"][0]['params']
|
| 45 |
+
# for k,v in applied_params.items():
|
| 46 |
+
# applied_params[k] = str(v)
|
| 47 |
+
# st.write(applied_params)
|
| 48 |
+
# st.write(data["replay"])
|
| 49 |
|
| 50 |
|
| 51 |
# show the images
|