自定义数据训练好模型之后，使用infer.py去预测，持续报错，始终找不到解决办法，求指导

/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/setuptools/depends.py:2: DeprecationWarning: the imp module is deprecated in favour of importlib; see the module's documentation for alternative uses
import imp
Traceback (most recent call last):
File "tools/infer.py", line 271, in
main()
File "tools/infer.py", line 106, in main
dataset.set_images(test_images)
AttributeError: 'VOCDataSet' object has no attribute 'set_images'

是什么原因导致没有该属性