IliaLarchenko commited on
Commit
ad0dde3
1 Parent(s): 9fce55b

tested replaycompose

Browse files
Files changed (1) hide show
  1. 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
- augmented_image = transform(image=image)["image"]
 
 
 
 
 
 
 
 
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