Skip to content

shasan7/DL_Fine_Tuning_MATLAB

Repository files navigation

Fine-Tune Pretrained Deep Learning Image Classification Models in MATLAB

While modifying these scripts for use, the dataset location is supposed to contain subfolders with folder names as labels of corresponding image categories. Due to different final layers structure for different networks, i had to provide separate codes for each of them. In my MATLAB version (R2023a), I can only save the model with "best-validation" where the "Metric' is automatically assumed validation loss. If you version is compatible, try "best-validation" with "accuracy" Metric in the trainingOptions for higher performance. If you require the cross-validation version of these codes, you'll find them in the "Cross Validation" folder. The corresponding MATLAB installation packages for pretrained networks are saved in "MATLAB_DL_Packages" folder.