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
sweecodengo is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
Check out our Code of Conduct.