garf 2.5__tar.gz → 2.6__tar.gz

{garf-2.5/garf.egg-info → garf-2.6}/PKG-INFO

@@ -1,17 +1,19 @@
 Metadata-Version: 2.1
 Name: garf
-Version: 2.5
+Version: 2.6
 Summary: Utility tools for GATE ARF simulations
-Home-page: https://github.com/dsarrut/garf
-Author: David Sarrut
-Author-email: david.sarrut@creatis.insa-lyon.fr
-License: UNKNOWN
-Platform: UNKNOWN
+Author-email: David Sarrut <david.sarrut@creatis.insa-lyon.fr>
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: itk
+Requires-Dist: numpy
+Requires-Dist: pre-commit
+Requires-Dist: torch>=2.1.0
+Requires-Dist: tqdm
 
 
 GARF = GATE ARF
@@ -27,5 +29,3 @@ https://www.ncbi.nlm.nih.gov/pubmed/30238925
 A method to speed up Monte-Carlo simulations of single photon emission computed tomography (SPECT) imaging is proposed. It uses an artificial neural network (ANN) to learn the angular response function (ARF) of a collimator-detector system. The ANN is trained once from a complete simulation including the complete detector head with collimator, crystal, and digitization process. In the simulation, particle tracking inside the SPECT head is replaced by a plane. Photons are stopped at the plane, and the energy and direction are used as input to the ANN, which provides detection probabilities in each energy window. Compared to histogram-based ARF, the proposed method is less dependent on the statistics of the training data, provides similar simulation efficiency, and requires less training data. The implementation is available within the GATE platform.
 
 - Gate10 test: https://github.com/OpenGATE/opengate/blob/master/opengate/tests/src/test043_garf.py
-
-

garf-2.6/garf/__init__.py

@@ -0,0 +1,4 @@
+# import files
+from .helpers import *
+from .garf_train import *
+from .garf_detector import *

garf-2.6/garf/bin/pytorch_info.py

@@ -0,0 +1,38 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import torch
+import click
+
+CONTEXT_SETTINGS = dict(help_option_names=["-h", "--help"])
+
+
+@click.command(context_settings=CONTEXT_SETTINGS)
+def pytorch_info():
+    """
+    \b
+    Print pytorch version
+    """
+
+    print("pytorch version", torch.__version__)
+
+    cuda_mode = torch.cuda.is_available()
+    if cuda_mode:
+        print("CUDA GPU mode is available")
+        print("CUDA version:", torch.version.cuda)
+        print("CUDA current device:", torch.cuda.current_device())
+        print("CUDA device counts:", torch.cuda.device_count())
+        print("CUDA device name:", torch.cuda.get_device_name(0))
+    else:
+        print("No CUDA GPU")
+
+    mps_mode = torch.backends.mps.is_available()
+    if mps_mode:
+        print("MPS GPU mode is available")
+    else:
+        print("No MPS GPU")
+
+
+# -----------------------------------------------------------------------------
+if __name__ == "__main__":
+    pytorch_info()