Skip to content

Repository files navigation

MRI-Retrieval

Retrieval of Brain MRI image

Preprocessing.ipynb --> Class including the following Functions :

image

Reorientation

image

Registration

image

Skull Stripping

image

Bias Field Correction and

image

Intensity Normalization

image

Overall work flow

image The Overall framework of our proposed approach. We feed a mini-batch of samples from the preprocessed image to the networks 𝑓𝑜 and 𝑓𝑡 , which are both based on the same architecture, and for the first network, weights and biases are calculated and binary hash based on these weights and biases calculated and stored in a mini-batch table (yellow shaded), and for the second network, creates an initial 𝑙 length weights to generate an initial binary hashes to be stored in the dictionary (blue shaded), later a new value is enqueued and an old value is dequeued; then the hash codes in the mini-batch and in the dictionary concatenated and form another dictionary with size of 𝑙 + 𝑞, for mini-batch size of 𝑞. At this stage, an online triplet mining is used to determine the triplet loss by mining a triplet samples. Cross-entropy loss also be calculated, summed with the triplet loss, and its gradient propagates back to the input layer of the network 𝑓𝑜 ; while for the network 𝑓𝑡 updated via a momentum strategy method. Upon completion of training, the retrieval set applied to the network and the output of the final layer of the network are stored in the database. Following this, the query image is applied to the same network and a hash code vector is generated; then a Hamming distance is computed between this hash code and the hash codes of retrieval sets that are saved in the database; finally, a list of the top N ranked images are returned as a retrieval result.

Training the model

Examples of input neuroimages, their corresponding extracted features and hash codes using handedness label for different 𝑘

image

Result - Example

Query image

image

Top nine retrieved output

image

Overall performance (Mean Average Precision - MAP)

image

About

Retrieval of Brain MRI image

Resources

Stars

0 stars

Watchers

2 watching

Forks

Releases

Packages

Contributors

Languages