`AttributeError: 'list' object has no attribute 'device'` about yolov9 HOT 17 CLOSED

#11 (comment) should be corrected.

    line 903 in utils/general.py 
    prediction = prediction[0][1]  # select only inference output, [0][0] for aux prediction, [0][1] for main prediction.

from yolov9.

So, the correct one is as follows, right?

 line 903 in utils/general.py 
    prediction = prediction[0][1]  

from yolov9.

Following this discussion, I opened a PR with a (probably temporal) fix. It makes detect.py work with both yolov9 and gelan models.

from yolov9.

    line 903 in utils/general.py 
    prediction = prediction[0][0]  # select only inference output

from yolov9.

@Dragos-Stan provides a temporal solution.

To inference yolov9 models, we need follow val_dual.py to construct detect_dual.py. Or do re-parameterization for yolov9 models, then using detect.py.

Currently, re-parametwrization part is under construction.

from yolov9.

So, the correct one is as follows, right?

 line 903 in utils/general.py 
    prediction = prediction[0][1]  

I followed yr ways, it worked with yolov9-c and yolov9-e.
But:

line 903 in utils/general.py 
     prediction = prediction[0] 

worked with gelan-c.pt and gelan-e.pt.

from yolov9.

Got same error in detect.py

from yolov9.

It works!

from yolov9.

I am getting the same error and fixed the general.py file as here discussions main...noorkhokhar99:yolov9:patch-1

from yolov9.

@SkalskiP Your fix is working for me for a yolov9-c model. Haven't tested ELAN.

Is this a better fix than @Dragos-Stan 's given it works for either architecture?

#11 (comment) should be corrected.

    line 903 in utils/general.py 
    prediction = prediction[0][1]  # select only inference output, [0][0] for aux prediction, [0][1] for main prediction.

from yolov9.

Hi @mkrupczak3 👋🏻 My fix should work with both GELAN and YOLOv9 architectures. Take a look here. I use it to run both models.

from yolov9.

@SkalskiP I found your fix applies to detect.py but does not fix val.py. Fixing utils/general.py according to WongKinYiu's comment fixes both however. I'm using that now

from yolov9.

@mkrupczak3 what's wrong with validation? I've done it yesterday and it works for me.

from yolov9.

Altering just detect.py does not fix val.py for dual models such as yolov9-c. Modifying utils/general.py as described does fix val.py however,

from yolov9.

    line 903 in utils/general.py 
    prediction = prediction[0][0]  # select only inference output

thanks

from yolov9.

    line 903 in utils/general.py 
    prediction = prediction[0][0]  # select only inference output

thanks

#11 (comment)

from yolov9.

Following this discussion, I opened a PR with a (probably temporal) fix. It makes work with both and models.detect.py``yolov9``gelan

thanks，its work

from yolov9.