mullenba

Results 10 comments of mullenba

Training on two consecutive images in the context of spatio-temporal learning

I'm trying to do something similar. Working with more than 3 channels is something of a mess apparently. It's not just the dataloader, the 3 channel limit is sprinkled throughout...

Training on two consecutive images in the context of spatio-temporal learning

I pulled down a fresh copy of the repo and moved in my custom dataloader. Adding "ch: 5" to the .yaml file did allow me to train 5 channel images...

Training on two consecutive images in the context of spatio-temporal learning

In my case, I can change it to this and make it work, but I think there's a bigger problem that I'm trying to pinpoint. I don't think the model...

Training on two consecutive images in the context of spatio-temporal learning

Ok, I'm looking at this line. https://github.com/ultralytics/yolov5/blob/e83b422a69bbd69628687b2dc50102c08877505c/models/common.py#L632 For my task, I'm passing the system numpy arrays of [960,960,5] and 12 classes. I printed the shape of y as it comes...

Training on two consecutive images in the context of spatio-temporal learning

I wrote a custom data loader that creates [960,960,5] arrays. It replaces `LoadImagesAndLabels`.

Training on two consecutive images in the context of spatio-temporal learning

@pourmand1376 Does it say it's training but you get 0 mAP after when you test? Did you check that your model is using all of the channels? Here's another point...

Training on two consecutive images in the context of spatio-temporal learning

Ok, when you're running the model for validation, how are you initializing it? For example, here's mine. ``` model = torch.hub.load('yolov5', # Folder with my customized repo 'custom', # This...

Training on two consecutive images in the context of spatio-temporal learning

You're taking a different approach to running the model than I am, so I'm not quite sure what the issue could be. Have you checked to see if it's getting...

Training on two consecutive images in the context of spatio-temporal learning

@kristinatel I created a custom dataset similar to [this](https://pytorch.org/tutorials/beginner/basics/data_tutorial.html#creating-a-custom-dataset-for-your-files). I then changed the utils.dataloaders file to call my new dataset here, where instead of calling LoadImagesAndLabels, I call my own...

Training on two consecutive images in the context of spatio-temporal learning

> The number of my data channels is 8. I replaced the dataset in yolov5 with a self-made dataset, and modified a few places where the error was reported because...