From 43291e42f10922ed52952ca4b8bfde834be72d84 Mon Sep 17 00:00:00 2001
From: alexcbb <alexchapin@hotmail.fr>
Date: Mon, 3 Jul 2023 10:07:54 +0200
Subject: [PATCH] Setup repo and change SAm
---
automatic_mask_train.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/automatic_mask_train.py b/automatic_mask_train.py
index 9e0ee59..0d0714f 100644
--- a/automatic_mask_train.py
+++ b/automatic_mask_train.py
@@ -91,7 +91,7 @@ if __name__ == '__main__':
images= []
from torchvision import transforms
transform = transforms.ToTensor()
- for j in range(10):
+ for j in range(2):
image = images_path[j]
img = cv2.imread(image)
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
--
GitLab