未验证 提交 082b1ac9 编写于 作者: W wangxinxin08 提交者: GitHub

fix transform problem of RandomCrop (#5412)

上级 b3c67db2
......@@ -1504,6 +1504,11 @@ class RandomCrop(BaseOperator):
if 'is_crowd' in sample:
sample['is_crowd'] = np.take(
sample['is_crowd'], valid_ids, axis=0)
if 'difficult' in sample:
sample['difficult'] = np.take(
sample['difficult'], valid_ids, axis=0)
return sample
return sample
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册