Generate Masks with OpenCV: 1_prepare_data.ipynb
Train UNet to cut dices from images: 2_train_unet.ipynb
Cut Images based on masks generated by OpenCV or UNet: 3_cut_images.ipynb
Classify single cutout dices using a CNN: 4_classify_single_dices.ipynb
install packages:
pip install --quiet torch torchvision pytorch-lightning wandb matplotlib pillow testresources pillow tqdm opencv-python imutils scikit-image
(and maybe jupyter if not run in CoCalc)