在线时间:8:00-16:00
迪恩网络APP
随时随地掌握行业动态
扫描二维码
关注迪恩网络微信公众号
开源软件名称:FashionAI开源软件地址:https://gitee.com/lmy0217/FashionAI开源软件介绍:FashionAIThis repo is code of FashionAI Global Challenge—Attributes Recognition of Apparel based on PyTorch. This repo only for learning. Environment
Prerequisites
UsageThe training and testing scripts come with several options, which can be listed with the python3 main.py --help To run the training and testing, simply run main.py. By default, the script runs resnet34 on attribute 'coat_length_labels' with 50 epochs. To training and testing resnet34 on attribute 'collar_design_labels' with 100 epochs and some learning parameters: python3 main.py --model 'resnet34' --attribute 'collar_design_labels' --epochs 100 --batch-size 128 --lr 0.01 --momentum 0.5 Every epoch trained model will be saved in the folder LicenseThe code is licensed with the MIT license. |
请发表评论