azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-22 12:26:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

GPU augmentation. try3. Failed.

Browse Source
This commit is contained in:
zxsanny
2025-03-05 15:32:19 +02:00
parent 2611a22264
commit 458d59753a
6 changed files with 240 additions and 316 deletions
Download Patch File Download Diff File Expand all files Collapse all files
preprocessing.py
+4 -5
View File
@@ -1,14 +1,13 @@
import concurrent.futures
import os.path
import time
from datetime import datetime, timedelta
from pathlib import Path
import albumentations as A
import cv2
import numpy as np
import concurrent.futures
from constants import (data_images_dir, data_labels_dir, processed_images_dir, processed_labels_dir,
annotation_classes, checkpoint_file, checkpoint_date_format)
from constants import (data_images_dir, data_labels_dir, processed_images_dir, processed_labels_dir)
from dto.imageLabel import ImageLabel
total_files_processed = 0
@@ -154,7 +153,7 @@ def main():
while True:
preprocess_annotations()
print('All processed, waiting for 2 minutes...')
time.sleep(120)
time.sleep(300)
if __name__ == '__main__':