Safety-Helmet-Wearing-Dataset icon indicating copy to clipboard operation
Safety-Helmet-Wearing-Dataset copied to clipboard

Published 20 hours ago • verified publisher icon njvisionpower

getModel()中classes的接收类型变成int类型了，原代码中list类型好像不匹配

Open FossilDaddy opened this issue 5 years ago • 1 comments

FossilDaddy avatar Feb 08 '20 06:02 FossilDaddy

net = get_model(net_name, pretrained_base=True) # , classes=len(classes) 去掉这个classes参数即可

gengyanlei avatar Mar 28 '20 01:03 gengyanlei