今天试用mmfashion 服装预测API进行了实验,从以下几点进行总结:
1.模型准备过程:
在这里把我遇到过的问题与大家分享,解决方案来自github官方页面的issue回答。
首先把环境搭好,可以参考我之前写过的一篇mmfashion环境准备
在兴致勃勃准备getting started时首先遇到了这个错误:
File ".../site-packages/mmfashion-0.4.0-py3.8.egg/mmfashion/models/builder.py", line 30, in _build_module
return obj_type(**args)
TypeError: __init()__ got an unexpected keyword argument 'loss_attr'
解决方案:在库文件(我的是复制在虚拟环境中的site-packages中的文件)中找到mmfashion/models/predictor/global_predictory.py
对里面的初始化__init()__
进行修改
def __init__(self,
backbone,
global_pool,
attr_predictor,
loss_attr, # 在这里添加一项
cate_predictor=None,
roi_pool=None,
pretrained=None):
接下来就是准备模型和预标注的数据Annotation
首先