cabouman / mbirjax

MBIRJAX is a Python package for Model Based Iterative Reconstruction (MBIR) of images from tomographic data.
https://mbirjax.readthedocs.io
BSD 3-Clause "New" or "Revised" License
7 stars 3 forks source link

Mar weight #42

Closed dyang37 closed 2 months ago

dyang37 commented 2 months ago

This PR contains the function to generate the weight for metal artifact reduction, along with a function to segment an image with Otsu's method.

Function interfaces: MAR weight: Screen Shot 2024-07-24 at 3 42 36 PM

Otsu's method image

Both the MAR weight and Otsu's method have been tested on two datasets (MAR dataset, AutoInjector).

To test:

  1. checkout branch mar_weight, and build mbirjax.
  2. Go to mbirjax_applications and checkout branch mar_weight.
  3. Run demo_nsi_mar.py. This script takes about 5 minutes to run on Gilbreth.
cabouman commented 2 months ago

I tried to run the test scripts, but got an error.

dyang37 commented 2 months ago

It seems that you are running the program on the front end of Gilbreth (gilbreth-fe00). You would need to reserve a computation node: sinteractive -A standby --time 4:00:00 -n 64 -N 1 --gpus-per-node=1 Then you may run the program on the node that you reserved.

cabouman commented 2 months ago

This ran fine, so I'm going to accept the PR to prerelease. We can discuss it tomorrow and see if it is ready for integration into the main branch.