解决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]
复现SSD+pytorch时会出现以上bug,错误原因来源于mode=random.choice(….)
解决方法,改为:
ran_num = np.random.choice([0,1,2,3,4,5])
mode=self.sample_options[ran_num]