IliaLarchenko
commited on
Commit
•
ad0dde3
1
Parent(s):
9fce55b
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
|