search

g-h-anna / ultralytics4channel

Ultralytics 8.2.5 modified to train and predict for 4-channel (RGBD) images.
GNU Affero General Public License v3.0
11 stars 0 forks source link

Colab modification #2

Open A7med01 opened 1 month ago

A7med01 commented 1 month ago

I am using colab how can i use your modified yolo ? can i lone your repo and install ?

A7med01 commented 1 month ago

I pulled original repo & installed it & cloned you repo & replaced your files and still can't train and got this error :

RuntimeError: Given groups=1, weight of size [16, 3, 3, 3], expected input[1, 4, 640, 640] to have 3 channels, but got 4 channels instead

g-h-anna commented 1 month ago

Hi, I am not sure how you can do this with Colab. My repository is tested for (1) installing the original ultralytics package (2) then chaging the files in the site-packages, in the "ultralytics" folder.

Based on your error message, I think still the 3 channel architecure is loaded, so not the source of my repo is used in the package content.

A7med01 commented 1 month ago

its is working with me now , it works only if i make train with device = cpu but when device = cuda it keep giving errors ,could you check this and tell me if it is same issue with you ?

A7med01 commented 2 weeks ago

any help