JiheonJeong commited on
Commit
f159474
·
1 Parent(s): 0f5564a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -144,15 +144,15 @@ def reset_textbox():
144
 
145
  def random_sample():
146
  global SAMPLE_IDX
147
- SAMPLE_IDX = get_random_sample(range(25), TOTAL_K)
148
- # SAMPLE_IDX = list(np.random.choice(range(25), TOTAL_K, replace = False))
149
  # SAMPLE_IDX = random.choices(range(25), k=TOTAL_K)
150
  image = np.array(Image.open(os.path.join(os.path.dirname(__file__), "data/Court.jpeg")))
151
  random_characters = [Image.open(os.path.join(os.path.dirname(__file__), f'data/characters_{idx:02}.png')).resize((64, 64)) for idx in SAMPLE_IDX]
152
  global CURRENT_POSITION
153
- CURRENT_POSITION = get_random_sample(RANDOM_POSITION, TOTAL_K)
154
- # CURRENT_POSITION = list(np.random.choice(np.array(RANDOM_POSITION), TOTAL_K, replace = False))
155
- # CURRENT_POSITION = random.choices(RANDOM_POSITION, k = TOTAL_K)
156
  new_image = Image.fromarray(image).convert('RGBA')
157
  for k in range(TOTAL_K):
158
  new_image.paste(random_characters[k], CURRENT_POSITION[k], random_characters[k])
 
144
 
145
  def random_sample():
146
  global SAMPLE_IDX
147
+ # SAMPLE_IDX = get_random_sample(range(25), TOTAL_K)
148
+ SAMPLE_IDX = list(np.random.choice(range(25), TOTAL_K, replace = False))
149
  # SAMPLE_IDX = random.choices(range(25), k=TOTAL_K)
150
  image = np.array(Image.open(os.path.join(os.path.dirname(__file__), "data/Court.jpeg")))
151
  random_characters = [Image.open(os.path.join(os.path.dirname(__file__), f'data/characters_{idx:02}.png')).resize((64, 64)) for idx in SAMPLE_IDX]
152
  global CURRENT_POSITION
153
+ # CURRENT_POSITION = get_random_sample(RANDOM_POSITION, TOTAL_K)
154
+ CURRENT_POSITION = [ RANDOM_POSITION[k] for k in list(np.random.choice(range(len(RANDOM_POSITION)), TOTAL_K, replace = False))]
155
+ # CURRENT_POSITION = random.choices(range(len(RANDOM_POSITION)), k = TOTAL_K)
156
  new_image = Image.fromarray(image).convert('RGBA')
157
  for k in range(TOTAL_K):
158
  new_image.paste(random_characters[k], CURRENT_POSITION[k], random_characters[k])