Spaces:
Running
Running
File size: 5,441 Bytes
87c3140 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 |
import os, pathlib, sys, inspect
from dataclasses import dataclass, field
currentdir = os.path.dirname(os.path.dirname(inspect.getfile(inspect.currentframe())))
parentdir = os.path.dirname(currentdir)
sys.path.append(parentdir)
sys.path.append(currentdir)
from vouchervision.general_utils import validate_dir, get_datetime
@dataclass
class Dir_Structure():
# Home
run_name: str = ''
dir_home: str = ''
dir_project: str = ''
# Processing dirs
path_archival_components: str = ''
path_config_file: str = ''
ruler_info: str = ''
ruler_overlay: str = ''
ruler_processed: str = ''
ruler_data: str = ''
ruler_class_overlay: str = ''
ruler_validation_summary: str = ''
ruler_validation: str = ''
save_per_image: str = ''
save_per_annotation_class: str = ''
binarize_labels: str = ''
# logging
path_log: str = ''
def __init__(self, cfg) -> None:
# Home
self.run_name = cfg['leafmachine']['project']['run_name']
self.dir_home = cfg['leafmachine']['project']['dir_output']
self.dir_project = os.path.join(self.dir_home,self.run_name)
validate_dir(self.dir_home)
self.__add_time_to_existing_project_dir()
validate_dir(self.dir_project)
# Processing dirs
self.path_archival_components = os.path.join(self.dir_project,'Archival_Components')
validate_dir(self.path_archival_components)
self.path_config_file = os.path.join(self.dir_project,'Config_File')
validate_dir(self.path_config_file)
self.path_cost = os.path.join(self.dir_project,'Cost')
validate_dir(self.path_cost)
# Logging
self.path_log = os.path.join(self.dir_project,'Logs')
validate_dir(self.path_log)
# self.custom_overlay_pdfs = os.path.join(self.dir_project,'Summary','Custom_Overlay_PDFs')
# self.custom_overlay_images = os.path.join(self.dir_project,'Summary','Custom_Overlay_Images')
###
# self.custom_overlay_pdfs = os.path.join(self.dir_project,'Summary','Custom_Overlay_PDFs')
# if cfg['leafmachine']['overlay']['save_overlay_to_pdf']:
# validate_dir(self.custom_overlay_pdfs)
# self.custom_overlay_images = os.path.join(self.dir_project,'Summary','Custom_Overlay_Images')
# if cfg['leafmachine']['overlay']['save_overlay_to_jpgs']:
# validate_dir(self.custom_overlay_images)
### Rulers
# self.ruler_info = os.path.join(self.dir_project,'Archival_Components','Ruler_Info')
# self.ruler_validation_summary = os.path.join(self.dir_project,'Archival_Components','Ruler_Info', 'Ruler_Validation_Summary')
# self.ruler_validation = os.path.join(self.dir_project,'Archival_Components','Ruler_Info', 'Ruler_Validation')
# self.ruler_processed = os.path.join(self.dir_project,'Archival_Components','Ruler_Info', 'Ruler_Processed')
# validate_dir(self.ruler_info)
validate_dir(os.path.join(self.path_archival_components, 'JSON'))
validate_dir(os.path.join(self.path_archival_components, 'labels'))
### Data
self.transcription = os.path.join(self.dir_project,'Transcription')
validate_dir(self.transcription)
self.transcription_ind = os.path.join(self.dir_project,'Transcription','Individual')
validate_dir(self.transcription_ind)
# self.transcription_ind_helper = os.path.join(self.dir_project,'Transcription','Individual_Helper_Content')
# validate_dir(self.transcription_ind_helper)
self.transcription_ind_OCR = os.path.join(self.dir_project,'Transcription','Individual_OCR')
validate_dir(self.transcription_ind_OCR)
self.transcription_ind_OCR_bounds = os.path.join(self.dir_project,'Transcription','Individual_OCR_Bounds')
validate_dir(self.transcription_ind_OCR_bounds)
self.transcription_ind_OCR_helper = os.path.join(self.dir_project,'Transcription','Individual_OCR_Helper')
validate_dir(self.transcription_ind_OCR_helper)
self.save_original = os.path.join(self.dir_project,'Original_Images')
validate_dir(self.save_original)
self.save_per_image = os.path.join(self.dir_project,'Cropped_Images', 'By_Image')
self.save_per_annotation_class = os.path.join(self.dir_project,'Cropped_Images', 'By_Class')
self.save_per_annotation_class = os.path.join(self.dir_project,'Cropped_Images', 'By_Class')
if cfg['leafmachine']['cropped_components']['save_per_image']:
validate_dir(self.save_per_image)
if cfg['leafmachine']['cropped_components']['save_per_annotation_class']:
validate_dir(self.save_per_annotation_class)
if cfg['leafmachine']['cropped_components']['binarize_labels']:
validate_dir(self.save_per_annotation_class)
# self.binarize_labels = os.path.join(self.dir_project,'Cropped_Images', 'By_Class','label_binary')
# validate_dir(self.binarize_labels)
def __add_time_to_existing_project_dir(self) -> None:
path = pathlib.Path(self.dir_project)
if path.exists():
now = get_datetime()
path = path.with_name(path.name + "_" + now)
self.run_name = path.name
path.mkdir()
self.dir_project = path
else:
path.mkdir()
self.dir_project = path |