解决SSD复现中的ValueError: setting an array element with a sequence错误

复现SSD+pytorch时会出现以上bug,错误原因来源于mode=random.choice(….)

 解决方法,改为:

ran_num = np.random.choice([0,1,2,3,4,5])
mode=self.sample_options[ran_num]

物联沃分享整理
物联沃-IOTWORD物联网 » 解决SSD复现中的ValueError: setting an array element with a sequence错误

发表回复