search

axondeepseg / sam_myelin_seg_TEM

Axon and Myelin segmentation using FAIR's Segment-Anything-Model (SAM)
MIT License
0 stars 0 forks source link

Add shuffling in dataloader #1

Closed hermancollin closed 1 year ago

hermancollin commented 1 year ago

https://github.com/axondeepseg/sam_myelin_seg_TEM/blob/415b98a0142b5dff71205749b641bc1e96638c84/scripts/bids_utils.py#L70-L87

Order in which samples are loaded should be shuffled.

hermancollin commented 1 year ago

Now supported with the custom AxonDataset class, which can be fed to torch.utils.data.DataLoader (see 5dfe03e)