sampling

pages/1_Omniart_Disentanglement.py

@@ -126,11 +126,16 @@ with input_col_2:
         
         choose_image_button = st.form_submit_button('Choose the defined image')
         random_id = st.form_submit_button('Generate a random image')
+        projection_id = st.form_submit_button('Generate an image on the boudary')
 
-        if random_id:
+        if random_id or projection_id:
             image_id = random.randint(0, 50000)
             st.session_state.image_id = image_id
             chosen_image_id_input.number_input('Image ID:', format='%d', step=1, value=st.session_state.image_id)
+            projection = False
+            
+            if projection_id:
+                projection = True
             
         if choose_image_button:
             image_id = int(image_id)
@@ -162,6 +167,9 @@ if st.session_state.space_id == 'Z':
 else:
     original_image_vec = annotations['w_vectors'][st.session_state.image_id]
 
+if projection:
+    original_image_vec = original_image_vec - np.dot(original_image_vec, separation_vector) * separation_vector
+
 img = generate_original_image(original_image_vec, model, latent_space=st.session_state.space_id)
 # input_image = original_image_dict['image']
 # input_label = original_image_dict['label']