search

zudi-lin / pytorch_connectomics

PyTorch Connectomics: segmentation toolbox for EM connectomics
http://connectomics.readthedocs.io/
MIT License
172 stars 77 forks source link

Add sample_label_size initialisation #21

Closed geekswaroop closed 4 years ago

geekswaroop commented 4 years ago

During inference on synapse detection, we will be loading the pre-trained model. The sample_label_size is not initialized globally leading to attribute errors.

zudi-lin commented 4 years ago

Thanks for finding the bug! The problem is defining label_vol_ratio when label is None. Therefore I put label_vol_ratio definition under 'if self.label is not None: '.