Commit 15a2f88f authored by Oksana Belyaeva's avatar Oksana Belyaeva
Browse files

refactoring

parent 101b7f49
......@@ -14,9 +14,6 @@ from config_project import CUR_DATA_DIR
DATASETNAME = 'train-0'
IMAGE_DIR = os.path.join(CUR_DATA_DIR, "images/" + DATASETNAME)
ANNOTATION_DIR = os.path.join(CUR_DATA_DIR, "csv")
OUTPUT_DIR = os.path.join(CUR_DATA_DIR, "jsons")
INFO = {
"description": CUR_DATA_DIR.split("/")[-1] + " Dataset",
......@@ -97,6 +94,10 @@ def convert_csv_to_coco():
}
image_id = 1
segmentation_id = 1
ANNOTATION_DIR = os.path.join(CUR_DATA_DIR, "csv")
IMAGE_DIR = os.path.join(CUR_DATA_DIR, "images/" + DATASETNAME)
OUTPUT_DIR = os.path.join(CUR_DATA_DIR, "jsons")
# filter for jpeg images
for root, _, files in os.walk(IMAGE_DIR):
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment