Deep-Learning-based-Layer-Prediction-Network Dependencies torch 1.8.0 python 3.8 skimage 0.19.2 numpy 1.23.1 opencv-python 4.6.0 timm 0.6.11 tqdm python benchmark/dataset.py --model model[ours] --path /where/is/your/dataset