mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 08:46:36 +00:00
remove fog and shadow augmentations
add install script
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user