Reaumur commited on
Commit
7a45f5b
·
verified ·
1 Parent(s): 38b1507

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -3
app.py CHANGED
@@ -37,6 +37,15 @@ img_width, img_height = 200, 50 # Ganti sesuai dimensi input gambar yang diguna
37
  # Definisikan max_length (misalnya panjang label maksimal)
38
  max_length = 50 # Ganti sesuai dengan panjang label teks maksimal yang diinginkan
39
 
 
 
 
 
 
 
 
 
 
40
  # Function to preprocess the image
41
  def prepare_image(img):
42
  # Resize gambar sesuai dengan ukuran yang diharapkan oleh model
@@ -63,11 +72,10 @@ def prepare_image(img):
63
  return pred_texts, preds
64
 
65
  def decode_batch_predictions(pred):
66
- # This function should convert the predictions (logits) to text
67
- # Modify this function based on your specific character map
68
  pred_texts = []
69
  for i in range(pred.shape[0]):
70
- pred_text = ''.join([characters[int(c)] for c in pred[i] if c != -1]) # Map to characters
 
71
  pred_texts.append(pred_text)
72
  return pred_texts
73
 
 
37
  # Definisikan max_length (misalnya panjang label maksimal)
38
  max_length = 50 # Ganti sesuai dengan panjang label teks maksimal yang diinginkan
39
 
40
+ # Pemetaan karakter yang mencakup huruf (kapital dan kecil) serta angka
41
+ characters = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
42
+ 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k',
43
+ 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v',
44
+ 'w', 'x', 'y', 'z', 'A', 'B', 'C', 'D', 'E', 'F', 'G',
45
+ 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R',
46
+ 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z']
47
+
48
+
49
  # Function to preprocess the image
50
  def prepare_image(img):
51
  # Resize gambar sesuai dengan ukuran yang diharapkan oleh model
 
72
  return pred_texts, preds
73
 
74
  def decode_batch_predictions(pred):
 
 
75
  pred_texts = []
76
  for i in range(pred.shape[0]):
77
+ # Ubah setiap prediksi menjadi daftar karakter
78
+ pred_text = ''.join([characters[int(c)] for c in pred[i] if c != -1 and c != 0]) # Pemetaan ke karakter
79
  pred_texts.append(pred_text)
80
  return pred_texts
81