paths.py

# -*- coding: utf-8 -*-
"""
Created on 13.01.20

"""
import os


class GeneralLoc:
    log_path = "/data/ssd4/ludwikat/logs"


class YTJumpLoc:
    base_path = '/data/ssd4/ludwikat/datasets/jump-broadcast'  # '<path_to_download_of>/jump-broadcast'
    pretrained_model = "pretrained_weights/coco_arbitrary_limbs.pth.tar"

    frames_path = "/data/ssd4/ludwikat/datasets/jump/youtube_dataset/annotated_frames"  # os.path.join(base_path,
    # 'annotated_frames')
    annotation_dir = 'keypoints'
    annotation_path = os.path.join(base_path, annotation_dir)
    segmentation_path = os.path.join(base_path, 'segmentations')
    segmentation_images = segmentation_path
    segmentation_bbox_path = os.path.join(base_path, 'bboxes_segmentations.csv')
    segmentation_endpoint_path = os.path.join(segmentation_path, 'endpoints_segmentations_{}{}.pkl')
    segmentation_anchor_path = os.path.join(segmentation_path, 'anchors_segmentations_{}.pkl')
    test_points = os.path.join(base_path, "test_points")