ToneZe

Results 5 comments of ToneZe

我也是出现相同问题,请问下原作者inference.py是预测部分吗?

The same problem. sys.platform: linux Python: 3.8.13 (default, Apr 19 2022, 00:53:22) [GCC 7.5.0] CUDA available: True GPU 0: Tesla V100-PCIE-16GB CUDA_HOME: /usr/local/cuda NVCC: Cuda compilation tools, release 10.2, V10.2.8...

File "/root/PyTorch_YOLOF/dataset/transforms.py", line 393, in __call__ image = F.resize(image, size=min_size, max_size=self.max_size) TypeError: resize() got an unexpected keyword argument 'max_size'

pip install -v -e . Using pip 23.0.1 from /usr/local/lib/python3.8/dist-packages/pip (python 3.8) Looking in indexes: https://mirrors.aliyun.com/pypi/simple Obtaining file:///hy-tmp/yolov5_obb/utils/nms_rotated Running command python setup.py egg_info running egg_info creating /tmp/pip-pip-egg-info-zf__u0d5/nms_rotated.egg-info writing /tmp/pip-pip-egg-info-zf__u0d5/nms_rotated.egg-info/PKG-INFO writing...