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

Revert "Add check.py"

Browse Source 
This reverts commit 6adc4c37e9.
This commit is contained in:
Oleksandr Bezdieniezhnykh
2024-05-26 23:10:25 +03:00
parent f463a6fcad
commit fdb784b392
21 changed files with 31 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+12 -30
View File
@@ -12,10 +12,9 @@ current_dataset_dir = os.path.join('datasets', 'zombobase-current')
def save(annotations, image, i, f_n):
cv2.imwrite(os.path.join(file_start_save, current_dataset_dir, f_n, images_dir, file_jpg[i]), image)
with open(os.path.join(file_start_save, current_dataset_dir, f_n, labels_dir, file_txt[i]), 'w') as f:
cv2.imwrite(os.path.join(file_start_save, current_dataset_dir, f_n, file_jpg[i]), image)
with open(os.path.join(file_start_save, current_dataset_dir, f_n, file_txt[i]), 'w') as f:
for iii in range(len(annotations)):
f.write(annotations[iii])
annotations = []
@@ -34,12 +33,7 @@ def piercing_photo_file():
def file_validation(annotations, file_txt, file_jpg):
try:
os.makedirs(os.path.join(file_start_save, current_dataset_dir,'validation'))
os.makedirs(os.path.join(file_start_save, current_dataset_dir, 'validation', images_dir))
os.makedirs(os.path.join(file_start_save, current_dataset_dir, 'validation', labels_dir))
except FileExistsError:
pass
os.makedirs(os.path.join(file_start_save, current_dataset_dir,'validation'))
for i in range(len(file_txt)):
image = cv2.imread(os.path.join(current_dataset_dir, images_dir, file_jpg[i]))
with open(os.path.join(current_dataset_dir, labels_dir, file_txt[i]), 'r') as file:
@@ -47,41 +41,29 @@ def file_validation(annotations, file_txt, file_jpg):
for line in lines:
annotations.append(line)
save(annotations, image, i, 'validation')
annotations = []
def sort_file():
try:
os.makedirs(os.path.join(file_start_save, current_dataset_dir))
except FileExistsError:
pass
os.makedirs(os.path.join(file_start_save, current_dataset_dir))
annotations = []
folder_name = ['test', 'train']
percent_file = [0.20, 0.10]
for f_n, p_f in zip(folder_name, percent_file):
try:
os.makedirs(os.path.join(file_start_save, current_dataset_dir, f_n))
os.makedirs(os.path.join(file_start_save, current_dataset_dir, f_n, images_dir))
os.makedirs(os.path.join(file_start_save, current_dataset_dir, f_n, labels_dir))
except FileExistsError:
pass
os.makedirs(os.path.join(file_start_save, current_dataset_dir, f_n))
for i in range(math.ceil(len(file_txt) * p_f)):
image = cv2.imread(os.path.join(current_dataset_dir, images_dir, file_jpg[i]))
with open(os.path.join(current_dataset_dir, labels_dir, file_txt[i]), 'r') as file:
lines = file.readlines()
for line in lines:
annotations.append(line)
save(annotations, image, i, f_n)
annotations = []
print(annotations)
file_txt.pop(i)
file_jpg.pop(i)
save(annotations, image, i, f_n)
file_txt.pop(i)
file_jpg.pop(i)
file_validation(annotations, file_txt, file_jpg)
def main():
piercing_photo_file()
sort_file()
sort_file()
main()