search

jianlin-cheng / CryoTransformer

A transformer model for picking protein particles in cryo-EM images
MIT License
33 stars 4 forks source link

Calculating performance metrics #1

Open ojasvijain opened 9 months ago

ojasvijain commented 9 months ago

Hi,

Great work on the code base!

I ran the predict.py on a few micrographs and wanted to know how I can compute the performance metrics such as precision, recall, f1 score.

I have the ground truth as bounding boxes around the particles (4 coordinates).

Thanks

ojasvijain commented 9 months ago

Hi, just a follow up.

I have managed to change my ground truth to coordinates (center) & radius - the same format as the output from CryoTransformer.

Is there any module to compute the metrics (precision & recall mainly)? If not, do you have any reference for the metrics you have published in the paper?

Thanks