yuragoithf commited on
Commit
8b7ee47
·
1 Parent(s): 67680bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -58,10 +58,10 @@ rows = 1
58
  columns = 1
59
 
60
  def gen_pred(img, model=seg_model):
61
- # pil_image = img.convert('RGB')
62
- # open_cv_image = np.array(pil_image)
63
- # img = open_cv_image[:, :, ::-1].copy()
64
- img = cv2.imread("./003e2c95d.jpg")
65
  img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]
66
  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
67
  img = img/255
 
58
  columns = 1
59
 
60
  def gen_pred(img, model=seg_model):
61
+ pil_image = img.convert('RGB')
62
+ open_cv_image = np.array(pil_image)
63
+ img = open_cv_image[:, :, ::-1].copy()
64
+ # img = cv2.imread("./003e2c95d.jpg")
65
  img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]
66
  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
67
  img = img/255