照片训练

参考该网站

https://auto.gluon.ai/stable/tutorials/image_prediction/kaggle.html

 

注册Kaggle

注册登录后下载:

 

 

下载地址如下:

https://www.kaggle.com/c/demo-shopee-iet-competition/data

下载的是一个结构目录,里边都是一些用照片训练的图片

 

 

整理好后新建一个main.py

from autogluon.vision import ImagePredictor, ImageDataset

if __name__ == '__main__':
    dataset = ImageDataset.from_folder('D:\py\data')
    predictor = ImagePredictor()
    predictor.fit(dataset, presets=['best_quality'])
    print('Top-1 val acc: %.3f' % predictor.fit_summary()['valid_acc'])

    # test_dataset = ImageDataset.from_folder('./data/shopeeiet/test')
    # pred = predictor.predict(test_dataset)
    # print(pred)

然后运行