future_data = pool.apply_async(
utils.raise_info_thread(next), (valid_loader, None))
for batch_idx in itertools.count():
curr_data = future_data
future_data = pool.apply_async(
utils.raise_info_thread(next), (valid_loader, None))
After Change
yolo_features_nnabla.d, args.conf_thresh, args.num_classes, args.anchors, args.num_anchors, 0, 1)
for i in range(yolo_features_nnabla.d.shape[0]):
if lineId >= total_samples:
print("Reached End of total_samples")
break
fileId = os.path.basename(valid_files[lineId]).split(".")[0]
width, height = utils.get_image_size(valid_files[lineId])
print(valid_files[lineId])