| 53 | |
| 54 | |
| 55 | def pack(): |
| 56 | with zipfile.ZipFile(OUTPUT_FILE, 'w', zipfile.ZIP_DEFLATED) as tf_model: |
| 57 | if os.path.isfile('dist/slim-0.1.tar.gz') and os.path.isfile('dist/object_detection-0.1.tar.gz'): |
| 58 | move_zip(tf_model, 'slim') |
| 59 | move_zip(tf_model, 'object_detection') |
| 60 | tf_model.write('bucket/prepare_data_object_detection.py') |
| 61 | tf_model.write('bucket/pipeline_skeleton.config') |
| 62 | tf_model.write('scripts/quick_export_graph.py') |
| 63 | |
| 64 | if os.path.isfile('dist/classification-0.1.tar.gz'): |
| 65 | move_zip(tf_model, 'classification') |
| 66 | tf_model.write('bucket/prepare_data_classification.py') |
| 67 | |
| 68 | tf_model.write('bucket/__init__.py') |
| 69 | tf_model.write('wml/__init__.py') |
| 70 | tf_model.write('wml/train_command.py') |
| 71 | tf_model.write('scripts/__init__.py') |
| 72 | tf_model.write('scripts/build_decoder.py') |
| 73 | tf_model.write('scripts/build_nms.py') |
| 74 | tf_model.write('scripts/convert_ssd_helper.py') |
| 75 | tf_model.write('scripts/convert_to_core_ml.py') |
| 76 | tf_model.write('scripts/convert_to_tfjs.py') |
| 77 | tf_model.write('scripts/convert_to_tflite.py') |
| 78 | tf_model.write('scripts/convert.py') |
| 79 | tf_model.write('scripts/types.py') |
| 80 | |
| 81 | |
| 82 | def setup_object_detection(): |