/vgg16-pytorch

vgg16 implemention by pytorch & transfer learning

Primary LanguagePython

vgg16-pytorch

vgg16 implemention by pytorch & transfer learning

enviroment

torch==1.2.0 torchvision==0.4.0

train_data & test_data

in the data file, save the training images, testing images and a label text file