File size: 459 Bytes
4a1f918 |
1 2 |
python generate_predictions.py --data_folder "/media/ubuntu/New Volume/jay/ISIC2018/ISIC2018_Task1-2_Test_Input" --data_config config_isic18.yml --model_config model_svdtuning.yml --save_path "svdtuning_shiftscale_isic2018_tal_focal075_alpha2_1e-3" --gt_path "/media/ubuntu/New Volume/jay/ISIC2018/ISIC2018_Task1_Test_GroundTruth" --labels_of_interest "Lesion" --pretrained_path "svdtuning_shiftscale_isic2018_tal_focal075_alpha2_1e-3.pth" --device "cuda:0"
|