File size: 3,548 Bytes
53bf77d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
import streamlit as st
from PIL import Image
from streamlit_drawable_canvas import st_canvas
import os
def image_annoter():
st.title("Image Annoter for YOLO")
st.write("Enter image files folder Location.")
# Upload video
file_location = st.text_input("Enter File folder Location",None)
if file_location is not None:
# Folder containing your images
label_folder = file_location
# Get a list of all images in the folder
image_files = [f for f in os.listdir(label_folder) if f.endswith(('png', 'jpg', 'jpeg'))]
# Initialize session state to keep track of the current image index
if 'current_index' not in st.session_state:
st.session_state.current_index = 0
# Set the downscaling factor
downscale_factor = 0.5 # Adjust the downscale factor as needed
# Navigation buttons
col1, col2, col3 = st.columns([1, 2, 1])
if col1.button("Previous"):
if st.session_state.current_index > 0:
st.session_state.current_index -= 1
if col3.button("Next"):
if st.session_state.current_index < len(image_files) - 1:
st.session_state.current_index += 1
# Display the current image
current_image_file = image_files[st.session_state.current_index]
st.write(f"Annotating: {current_image_file}")
image_path = os.path.join(label_folder, current_image_file)
image = Image.open(image_path)
# Downscale the image for the canvas
scaled_width = int(image.width * downscale_factor)
scaled_height = int(image.height * downscale_factor)
scaled_image = image.resize((scaled_width, scaled_height))
# Display the image on the canvas
canvas_result = st_canvas(
fill_color="rgba(255, 0, 0, 0.3)", # Fill color for the bounding box
stroke_width=3,
stroke_color="#ff0000",
background_image=scaled_image,
height=scaled_height,
width=scaled_width,
drawing_mode="rect",
key=current_image_file
)
# Save annotations
if st.button("Save Annotation"):
if canvas_result.json_data is not None:
# Extract coordinates of the bounding box
for obj in canvas_result.json_data["objects"]:
if obj["type"] == "rect":
left = obj["left"] / downscale_factor
top = obj["top"] / downscale_factor
width = obj["width"] / downscale_factor
height = obj["height"] / downscale_factor
x_center = left + width / 2
y_center = top + height / 2
# Normalize the coordinates (YOLO format)
x_center /= image.width
y_center /= image.height
width /= image.width
height /= image.height
# Save the annotation to a .txt file
annotation_path = os.path.join(label_folder, current_image_file.replace('.jpg', '.txt').replace('.png', '.txt'))
with open(annotation_path, 'w') as f:
f.write(f"0 {x_center} {y_center} {width} {height}\n")
st.success(f"Annotation saved for {current_image_file}")
|