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

Revert "Made some changes"

Browse Source 
This reverts commit 8529bd0c6f.
This commit is contained in:
Oleksandr Bezdieniezhnykh
2024-05-26 23:10:28 +03:00
parent fdb784b392
commit fb7ccdd2d8
2 changed files with 2 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files
preprocessing.py
+2 -1
View File
@@ -42,6 +42,7 @@ def image_processing(img_ann: ImageAnnotation) -> [ImageAnnotation]:
transform = A.Compose([
A.HorizontalFlip(p=0.5),
A.RandomBrightnessContrast(p=0.2),
A.ShiftScaleRotate(shift_limit=0.1, scale_limit=0.2, rotate_limit=15, p=0.5),
], bbox_params=A.BboxParams(format='yolo', label_fields=['category_ids']))
bboxes = bboxes
@@ -55,7 +56,7 @@ def image_processing(img_ann: ImageAnnotation) -> [ImageAnnotation]:
return transformed_image, transformed_bboxes, transformed_category_ids
def write_results(img_ann: ImageAnnotation):
for i in range(100):
for i in range(10):
transformed_image, transformed_bboxes, transformed_category_ids, = image_processing(img_ann)
cv2.imwrite(os.path.join(current_dataset_dir, images_dir, str(i)+ImageAnnotation(img_ann).image_path + '.jpg'), transformed_image)
with open(os.path.join(current_dataset_dir, labels_dir, str(i)+ImageAnnotation(img_ann).image_path + '.txt'), 'w') as f: