initial commit

Author: gymicv

README.md

@@ -16,8 +16,8 @@
   </h3>
 </div>
 
- ![STR](https://github.com/mineeuk/PETITE/assets/72694034/2641a7f3-facc-4eac-84cf-b96ea3c32f64)
- 
+<img width="1194" alt="str" src="https://github.com/user-attachments/assets/07e49bf9-a8ca-4e81-ab5d-b97bce895875">
+
 
 ### Requirements
 - [3D CVT-GAN](https://github.com/Aru321/GAN-for-PET-Reconstruction)
@@ -81,4 +81,4 @@ import loralib as lora
 ### ADNI policy
 ADNI data cannot be publicly shared by private users. It was not possible to directly upload data in this repository. However, feel free to [contact me]([email protected]) if you need more information about how to access data or about data used.
 
-Remember to follow the ADNI data sharing and publication policy, if you are going to publish a work in which data from the ADNI database were used.
+Remember to follow the ADNI data sharing and publication policy, if you are going to publish a work in which data from the ADNI database were used.

shell/train/petite.sh

@@ -52,7 +52,7 @@ for json in "${jsons[@]}"; do
                     --logdir "" \
                     --pretrained_dir "./base_raw/Resolution/${folder}#${json}/" \
                     --pretrained_model_name "./base_raw/Resolution/final_${folder}#${json}.pt" \\
-                    --csv_dir "./csv_result/${type}/num_${num}.xlsx"\
+                    --csv_dir "/root_dir/csv/${type}/num_${num}.xlsx"\
                     --optim_lr "1e-3" \
                     --reg_weight "1e-5" \
                     --optim_name "adamw" \

visualization.py

@@ -19,13 +19,13 @@
 parser.add_argument("--label_path", default="/root_dir/ADNI/Dynamic/Resolution/labels/", type=str, help="label folder path (GT)")
 
 parser.add_argument("--base_path", default="", type=str, help="base model results folder path (CVTGAN)")
-parser.add_argument("--fft_path", default="", type=str, help="VPT-shallow results folder path")
-parser.add_argument("--ln_path", default="", type=str, help="VPT-shallow results folder path")
-parser.add_argument("--bias_path", default="", type=str, help="VPT-deep results folder path")
-parser.add_argument("--lora_path", default="", type=str, help="VPT-shallow results folder path")
-parser.add_argument("--adapter_path", default="", type=str, help="VPT-deep results folder path")
-parser.add_argument("--ssf_path", default="", type=str, help="VPT-shallow results folder path")
-parser.add_argument("--shallow_path", default="", type=str, help="VPT-deep results folder path")
+parser.add_argument("--fft_path", default="", type=str, help="full fine tuning results folder path")
+parser.add_argument("--ln_path", default="", type=str, help="layernorm results folder path")
+parser.add_argument("--bias_path", default="", type=str, help="bitfit results folder path")
+parser.add_argument("--lora_path", default="", type=str, help="lora results folder path")
+parser.add_argument("--adapter_path", default="", type=str, help="adapters results folder path")
+parser.add_argument("--ssf_path", default="", type=str, help="ssf results folder path")
+parser.add_argument("--shallow_path", default="", type=str, help="VPT-shallow results folder path")
 parser.add_argument("--our_path", default="", type=str, help="VPT-deep results folder path")
 
 parser.add_argument("--save_path", default="./result/source_target/save_result/", type=str, help="The folder path to save")