add train.py

form dataset for current date
add exception catching
This commit is contained in:
Oleksandr Bezdieniezhnykh
2024-06-05 23:35:06 +03:00
parent 3fd726f9c7
commit 07ea67746a
7 changed files with 168 additions and 26 deletions
+4 -7
View File
@@ -1,19 +1,16 @@
from pathlib import Path
import cv2
import os.path
from dto.annotationClass import AnnotationClass
from dto.imageLabel import ImageLabel
from preprocessing import read_labels
images_dir = '../images'
labels_dir = '../labels'
annotation_classes = AnnotationClass.read_json()
images_dir = ''
image = os.listdir(images_dir)[0]
image_path = os.path.join(images_dir, image)
labels_path = os.path.join(labels_dir, f'{Path(image_path).stem}.txt')
image_path = 'test01.jpg'
labels_path = 'test01.txt'
img = ImageLabel(
image_path=image_path,
BIN
View File
Binary file not shown.

After

Width:  |  Height:  |  Size: 105 KiB

+1
View File
@@ -0,0 +1 @@
0 0.3809 0.49269 0.21636 0.39129