Omnibus commited on
Commit
0680ac2
Β·
1 Parent(s): 026d9f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -13
app.py CHANGED
@@ -183,6 +183,7 @@ def blur_im(img,bounds,target_lang,trans_lang):
183
  im = cv2.imread(img)
184
  im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
185
  for bound in bounds:
 
186
  p0, p1, p2, p3 = bound[0]
187
  x = int(p0[0])
188
  y = int(p0[1])
@@ -191,21 +192,21 @@ def blur_im(img,bounds,target_lang,trans_lang):
191
  kernel = np.ones((3, 3), np.uint8)
192
  im[y:y+h, x:x+w] = cv2.dilate(im[y:y+h, x:x+w], kernel, iterations=3)
193
  im[y:y+h, x:x+w] = cv2.GaussianBlur(im[y:y+h, x:x+w],(51,51),0)
194
-
195
-
196
-
197
  im = Image.fromarray(im)
198
  for bound in bounds:
199
- p0, p1, p2, p3 = bound[0]
200
- x = int(p0[0])
201
- y = int(p0[1])
202
- w = int(p2[0]) - int(x)
203
- h = int(p2[1]) - int(y)
204
- draw = ImageDraw.Draw(im)
205
- font_size=int(int(w)*0.2)
206
- font = ImageFont.truetype("./fonts/unifont-15.0.01.ttf", int(font_size))
207
- text = this(bound[1],target_lang,trans_lang)
208
- draw.text((x, y),text, font = font, fill=(0,0,0))
 
 
 
209
  return im
210
 
211
  def draw_boxes(image, bounds, color='blue', width=1):
 
183
  im = cv2.imread(img)
184
  im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
185
  for bound in bounds:
186
+
187
  p0, p1, p2, p3 = bound[0]
188
  x = int(p0[0])
189
  y = int(p0[1])
 
192
  kernel = np.ones((3, 3), np.uint8)
193
  im[y:y+h, x:x+w] = cv2.dilate(im[y:y+h, x:x+w], kernel, iterations=3)
194
  im[y:y+h, x:x+w] = cv2.GaussianBlur(im[y:y+h, x:x+w],(51,51),0)
 
 
 
195
  im = Image.fromarray(im)
196
  for bound in bounds:
197
+ if bound[2]>=0.3:
198
+ p0, p1, p2, p3 = bound[0]
199
+ x = int(p0[0])
200
+ y = int(p0[1])
201
+ w = int(p2[0]) - int(x)
202
+ h = int(p2[1]) - int(y)
203
+ draw = ImageDraw.Draw(im)
204
+ font_size=int(int(w)*0.15)
205
+ font = ImageFont.truetype("./fonts/unifont-15.0.01.ttf", int(font_size))
206
+ text = this(bound[1],target_lang,trans_lang)
207
+ draw.text((x, y),text, font = font, fill=(0,0,0))
208
+ else:
209
+ pass
210
  return im
211
 
212
  def draw_boxes(image, bounds, color='blue', width=1):