Open AkankshaP0102 opened 5 months ago
You replace my loader wirh your dataloader based on pytorch custom dataloader!
We did not support video segmentation!
Thank you for your quick response @ByungKwanLee . Is there any way I can implement Video segmentation using Causal since the architecture performs quite well on images??
Dealing with each image frame extracted from video may be a proper trial
Thank you.Will work on it.
Working on training the model on custom dataset but not getting any luck on creating the dataloader. Please can you help @ByungKwanLee as in how can I create my own dataloader as I am new in the field of computer vision task. Thank you.
Did you see Pytorch Custom Dataloader documentation?
yes sir created my own pytorch custom dataloader but facing issue while running the step 2 of training process. The error looks like this: File "/media/emsg/2d46715b-293d-4478-acd4-5f000d443896/anaconda3/envs/causal/lib/python3.9/site-packages/torch/multiprocessing/spawn.py", line 68, in _wrap fn(i, args) File "/media/emsg/2d46715b-293d-4478-acd4-5f000d443896/Causal-Unsupervised-Segmentation/train_front_door_mlp.py", line 263, in main train( File "/media/emsg/2d46715b-293d-4478-acd4-5f000d443896/Causal-Unsupervised-Segmentation/utils/utils.py", line 36, in wrapper func(args, **kwards) File "/media/emsg/2d46715b-293d-4478-acd4-5f000d443896/Causal-Unsupervised-Segmentation/train_front_door_mlp.py", line 87, in train loss_linear = F.cross_entropy(flat_linear_logits[flat_label_mask], flat_label[flat_label_mask]) IndexError: The shape of the mask [2408448] at index 0 does not match the shape of the indexed tensor [802816, 39] at index 0 Please help me with the same. your help will be be appreciated. Thank you in advance
How can I use Causal for training on custom dataset? Is it possible to use it for video semantic segmentation??