Timor

Thisi is my calibrator.py with imagestream ``` import os import cv2 import numpy as np import tensorrt as trt import pycuda.autoinit # import pycuda.driver as cuda class PythonEntropyCalibrator(trt.IInt8EntropyCalibrator2): def __init__(self,...

create_calibration_dataset2 - this function returns a list of paths (full paths to image files - the calibration set) for example: ``` def create_calibration_dataset(data_dir, max_len=100): """Create list of calibration images (filename)...

Thenk you for openning this issue. I see the problem, for now I will recommand you to do the dwt for each channel separatly. I will work on a fix...

Thanks, I added a fix for the shape issue and added a test for it :) https://github.com/Timorleiderman/tensorflow-wavelets/blob/35d9f81286f6808483d7b2edf6f6178c73b1b826/src/test.py#L107

/review