1. Anuncie Aqui ! Entre em contato fdantas@4each.com.br

[Python] How do I send batch images to the Custom Vision service with there associated label...

Discussão em 'Python' iniciado por Stack, Setembro 13, 2024.

  1. Stack

    Stack Membro Participativo

    I've been working on trying to send my training data to the custom vision service for a couple hours now and keep running into issue after issue. I have a large dataset of 430 images and a accompanying csv containing the associated tags for each(using multi-tag on custom vision). I've spend hours doing research and eventually go the point where the API and SDKs are correctly linked and everything's setup and now keep getting the error "Batch upload failed: 'Image' object has no attribute 'error'".

    here is the image processing and batch upload section of my code:

    batch_size = 64
    publish_iteration_name = "classifyModel"

    # 6. Open CSV and Process Images
    with open(image_data_file, "r") as csvfile:
    reader = csv.DictReader(csvfile)
    image_batch = []

    for row in reader:
    image_path = os.path.join(image_folder, row["filename"])
    if not os.path.exists(image_path):
    print(f"Warning: Image not found: {image_path}")
    continue

    with open(image_path, "rb") as image_contents:
    image_data = image_contents.read()
    tag_ids = []

    for key, value in row.items():
    if key != "filename" and value == "1":
    try:
    tag = trainer.get_tag(PROJECT_ID, key)
    tag_ids.append(tag.id)
    except Exception as e:
    print(f"Error getting tag '{key}': {e}")

    # Add Image File to the Batch
    image_batch.append(ImageFileCreateEntry(
    name=image_path, contents=image_data, tag_ids=tag_ids))

    # Upload Batch if it reaches the size limit
    if len(image_batch) >= batch_size:
    try:
    upload_result = trainer.create_images_from_files(
    PROJECT_ID, ImageFileCreateBatch(images=image_batch))

    if not upload_result.is_batch_successful:
    for image in upload_result.images:
    if image.status != "OK":
    print(
    # Check error details inside image.image
    f"Image '{image.source_url}' upload failed. Status: {image.status}, Error Details: {image.image.error.message if image.image.error else 'Unknown'}"
    )
    else:
    print(
    f"Image '{image.source_url}' uploaded successfully with ID: {image.image.id}")

    except Exception as e:
    print(f"Batch upload failed: {e}")


    my program also handles training but I have been unsuccessful getting to this point as I continue to not have enough images to train. Thanks to anyone who is willing to help out!.

    I've scoured the documentation and forums trying to gain insight on what the issue might be(I have a feeling its something simple) and I'm now at a road block. any suggestions are welcome haha

    Continue reading...

Compartilhe esta Página