Clothing Image Classification with DenseNet201 Network and Optimized Regularized Random Vector Functional Link
Zhiyu Zhou,Mingxuan Liu,Wenxiong Deng,Yaming Wang,Zefei Zhu
DOI: https://doi.org/10.1080/15440478.2023.2190188
2023-03-28
Journal of Natural Fibers
Abstract:To ameliorate the precision of clothing image classification, we proposed a clothing image classification method via the DenseNet201 network based on transfer learning and the optimized regularized random vector functional link (RVFL). First, the formula extracts weight's parameters about DenseNet201 that is pre-trained on the ImageNet dataset for transfer learning, thereby obtaining an incipient network,after that trim this model parameters. The modified network is utilized to pick up the clothing image features output by the DenseNet201's global average pooling layer. Second, regularization coefficient is introduced to control RVFL's model complexity and solve the problem of over-fitting. Then, the generated solution vector of aquila optimizer (AO) is produced by marine predators algorithm (MPA). The input weights, biases of hidden layer and renormalization modulus of regularized RVFL are optimized using the improved AO algorithm. Finally, we use the optimized RVFL to assort abstracted fashion graphics traits. We use Accuracy, Macro-F1, Macro-R and Macro-P to assess the algorithm's ability and compare this algorithm with ResNet50 network, ResNet101 network, DenseNet201 network, InceptionV3 network and different classifiers, which use DenseNet201 as the feature extractor to get the input. From the experimental results, this algorithm proposed has excellent classification power and generalization ability.
materials science, textiles