Added --download_weights option and fixed setup.py

nextqsm/predict_all.py

@@ -6,6 +6,8 @@
 
 import argparse
 import os
+import sys
+
 from pathlib import Path
 
 def main(ckp_path, ckp_name, source_path, mask_path, out_file, b_vec):
@@ -97,6 +99,14 @@ def download_weights():
 
 
 def cli_main():
+    if '--download_weights' in sys.argv:
+        parser = argparse.ArgumentParser(description="Download weights for NeXtQSM.")
+        parser.add_argument('--download_weights', action='store_true', help='Only download the weights and exit')
+        args = parser.parse_args()
+
+        if args.download_weights:
+            download_weights()
+            exit()
 
     parser = argparse.ArgumentParser(
         description="NeXtQSM: Deep Learning QSM Algorithm",

setup.py

@@ -3,6 +3,7 @@
 
 setup(
     name="nextqsm",
+    long_description="NeXtQSM is a complete deep learning pipeline for data-consistent quantitative susceptibility mapping trained with hybrid data",
     version="1.0.2",
     packages=find_packages(),
     package_dir={'nextqsm': 'nextqsm'},