hopeu

Batch inference problem!

1

I retrained the model with 16 batches. Then I converted to ONNX. **ONNX Sample** ort_session = ort.InferenceSession("yunet_16.onnx", providers=['CUDAExecutionProvider']) input_name = ort_session.get_inputs()[0].name for i_path in tqdm(os.listdir(images_path)): for i in os.listdir(os.path.join(images_path, i_path)):...