search

aws-samples / amazon-sagemaker-tensorflow-object-detection-api

Train and deploy models using TensorFlow 2 with the Object Detection API on Amazon SageMaker
MIT No Attribution
44 stars 34 forks source link

Update prepare_data.ipynb #4

Closed aadu999 closed 3 years ago

aadu999 commented 3 years ago

Labeling Job Argument added Issues 

Traceback (most recent call last):
  File "/opt/program/prepare_data.py", line 30, in <module>
    tf_record_generator.generate_tf_records()
  File "/opt/program/utils/tf_record_util.py", line 30, in generate_tf_records
    annotation_dict[self.label_job]['annotations'])
KeyError: None

Description of changes: The document was missing the labeling job argument for generating TF Record and It was showing an error. Fixed it by adding 

f'--labeling_job={labeling_job_name}'