thank you for your answer.
i had tried it, with the torchvision model alexnet, but it got 1000 classes, so the classification is not  efficient. could i build and train my own model, and save the weights in order to use the pre-trained model ??