remove fog and shadow augmentations

add install script
This commit is contained in:
zxsanny
2025-05-19 08:30:25 +03:00
parent 292809593c
commit 96056f53ad
5 changed files with 19 additions and 17 deletions
+4 -5
View File
@@ -38,7 +38,6 @@ DEFAULT_CLASS_NUM = 80
total_files_copied = 0
def form_dataset(from_date: datetime):
makedirs(today_dataset, exist_ok=True)
images = []
old_images = []
@@ -180,10 +179,10 @@ def train_dataset(existing_date=None, from_scratch=False):
cur_folder = f'{prefix}{existing_date}'
cur_dataset = path.join(datasets_dir, f'{prefix}{existing_date}')
else:
if from_scratch:
shutil.rmtree(today_dataset)
form_dataset(latest_date)
create_yaml()
# if from_scratch and Path(today_dataset).exists():
# shutil.rmtree(today_dataset)
# form_dataset(latest_date)
# create_yaml()
cur_folder = today_folder
cur_dataset = today_dataset