MobileNet多标签分类
安装环境依赖
。
pip install numpy scikit-image scipy scikit-learn matplotlib tensorboard -i https://pypi.tuna.tsinghua.edu.cn/simple
PyTorch-Multi-Label-Image-Classification-Mobilenet
基于mobilenet v3的图像多标签分类 mobilenet
。
python train.py --attributes_file ./fashion-product-images/styles.csv --device cuda