fix install_success problem for batch

2023-06-02 22:57:20 +08:00 · 2023-06-02 22:57:20 +08:00 · b2ab5f6f0f
parent 76c44793cd
commit b2ab5f6f0f
1 changed files with 1 additions and 0 deletions
--- a/scripts/sam.py
+++ b/scripts/sam.py
@ -258,6 +258,7 @@ def dino_batch_process(
    predictor = SamPredictor(sam)
    
    process_info = ""
+    install_success = True
    all_files = glob.glob(os.path.join(dino_batch_source_dir, "*"))
    for image_index, input_image_file in enumerate(all_files):
        print(f"Processing {image_index}/{len(all_files)} {input_image_file}")