kamangir commited on
Commit
a1a8891
1 Parent(s): 60603bc

validating fashion_mnist train - kamangir/bolt#689

Browse files
image_classifier/__init__.py CHANGED
@@ -1,5 +1,5 @@
1
  name = "image_classifier"
2
 
3
- version = "1.1.64"
4
 
5
  description = "fashion-mnist + hugging-face + awesome-bash-cli"
 
1
  name = "image_classifier"
2
 
3
+ version = "1.1.65"
4
 
5
  description = "fashion-mnist + hugging-face + awesome-bash-cli"
image_classifier/classes.py CHANGED
@@ -155,18 +155,16 @@ class Image_Classifier(object):
155
  max_index = test_images.shape[0]
156
  if page_count != -1:
157
  max_index = min(24 * page_count, max_index)
158
- offset = int(np.max(np.array(objects.list_of_frames(output_path) + [-1]))) + 1
159
- logger.info(
160
- "image_classifier.predict(offset={}): rendering {} frame(s)...".format(
161
- offset, max_index
162
- )
163
- )
164
  for index in tqdm(range(0, max_index, 24)):
165
  self.render(
166
  predictions[index : index + 24],
167
  None if test_labels is None else test_labels[index : index + 24],
168
  test_images[index : index + 24],
169
- "{}/Data/{}/info.jpg".format(output_path, int(index / 24) + offset),
 
 
 
170
  prediction_time,
171
  )
172
 
 
155
  max_index = test_images.shape[0]
156
  if page_count != -1:
157
  max_index = min(24 * page_count, max_index)
158
+ logger.info(f"image_classifier.predict(): rendering {max_index} frame(s)...")
 
 
 
 
 
159
  for index in tqdm(range(0, max_index, 24)):
160
  self.render(
161
  predictions[index : index + 24],
162
  None if test_labels is None else test_labels[index : index + 24],
163
  test_images[index : index + 24],
164
+ "{}/image_classifier/prediction_output/{:05d}.jpg".format(
165
+ output_path,
166
+ int(index / 24),
167
+ ),
168
  prediction_time,
169
  )
170