Update pages/Data Collection.py

pages/Data Collection.py

@@ -462,6 +462,32 @@ elif st.session_state.current_page == "explore_images_video":
     RGB is the most common color space used in digital imaging and is ideal for screens and devices that emit light.
     """)
 
+    st.subheader("Visualization of RGB Channels")
+    
+    # Load an example image (you can use any image)
+    uploaded_file = st.file_uploader("Upload an Image", type=["jpg", "png"])
+    if uploaded_file:
+        img = Image.open(uploaded_file)
+        img_array = np.array(img)
+    
+        # Split into channels
+        red_channel = img_array[:, :, 0]
+        green_channel = img_array[:, :, 1]
+        blue_channel = img_array[:, :, 2]
+    
+        # Plot the channels
+        fig, ax = plt.subplots(1, 4, figsize=(12, 4))
+        ax[0].imshow(img)
+        ax[0].set_title("Original Image")
+        ax[1].imshow(red_channel, cmap="Reds")
+        ax[1].set_title("Red Channel")
+        ax[2].imshow(green_channel, cmap="Greens")
+        ax[2].set_title("Green Channel")
+        ax[3].imshow(blue_channel, cmap="Blues")
+        ax[3].set_title("Blue Channel")
+        for a in ax:
+            a.axis("off")
+        st.pyplot(fig)