truens66 commited on
Commit
2d5fbc5
·
verified ·
1 Parent(s): 8ed8edd

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +10 -0
pipeline.py CHANGED
@@ -1,6 +1,16 @@
1
  import cv2
2
  import numpy as np
3
  import tensorflow as tf
 
 
 
 
 
 
 
 
 
 
4
 
5
  # Load model
6
  model = tf.keras.models.load_model("efficientnet-b0/")
 
1
  import cv2
2
  import numpy as np
3
  import tensorflow as tf
4
+ import zipfile
5
+
6
+ # Set random seed for reproducibility.
7
+ tf.random.set_seed(42)
8
+
9
+ # Extract model if zipped.
10
+ local_zip = "./efficientnet-b0.zip"
11
+ zip_ref = zipfile.ZipFile(local_zip, 'r')
12
+ zip_ref.extractall()
13
+ zip_ref.close()
14
 
15
  # Load model
16
  model = tf.keras.models.load_model("efficientnet-b0/")