The winning solution of the Endocv-2021 grand challange
Endocv2021-winner This is the winning solution of the Endocv-2021 grand challange. Dependencies pytorch # tested with 1.7 and 1.8 torchvision tqdm pandas numpy albumentations # for augmentations torchsummary segmentation_models_pytorch # for basic segmentaion models pyra_pytorch # pyra_pytorch.PYRADatasetFromDF is used. But this can be replaced with normal pytorch dataset. Tri-Unet Block diagram of Tri-Unet How to train Tri-Unet and other basic models to DivergentNet? # To train Tri-unet python tri_unet.py train –num_epochs 2 –device_id 0 –train_CSVs sample_CSV_files/C1.csv sample_CSV_files/C1.csv –val_CSVs sample_CSV_files/C2.csv sample_CSV_files/C3.csv […]
Read more