Preface: using yolov5-6.0 version, you want to detect several pictures, but there is a problem in the title. It can be seen that the upsampling function is not quite right. Now record the solution.
Version: yolov5-6.0, python3.8, pytorch1.11.0
1. Problem recurrence
2. Official website solution
This problem first appeared in yolov5 and is related to pytoch 1.11.0.
In other words, this problem may be encountered in both train and detect. The following is the solution to reduce the pytoch version to less than 10.
Then the blogger made a fix for PyTorch version 1.11.0
But it doesn’t seem to be solved. The my torch version is 1.11.0, but this problem still occurs.
The solution of modifying the upper sampling function given by netizens.
Just Comment out this part below.
It’s really solved.