File postprocess_results.py has 264 lines of code (exceeds 250 allowed). Consider refactoring.

Open

#!/usr/bin/env python

"""
Post-processes WormPose results by interpolating over missing frames and smoothing
"""

Function _compare_pred_real has a Cognitive Complexity of 17 (exceeds 5 allowed). Consider refactoring.

Open

def _compare_pred_real(
    frame_preprocessing: BaseFramePreprocessing,
    data_reading_queue,
    results_queue,
    image_shape,

• Read up
  Read up

Function load_dataset has a Cognitive Complexity of 13 (exceeds 5 allowed). Consider refactoring.

Open

def load_dataset(
    dataset_loader: str,
    dataset_path: str,
    selected_video_names: Optional[List[str]] = None,
    resize_options: ResizeOptions = None,

• Read up
  Read up

Function _parse_arguments has a Cognitive Complexity of 12 (exceeds 5 allowed). Consider refactoring.

Open

def _parse_arguments(dataset_path: str, kwargs: dict):
    if kwargs.get("work_dir") is None:
        kwargs["work_dir"] = default_paths.WORK_DIR
    if kwargs.get("num_process") is None:
        kwargs["num_process"] = os.cpu_count()

• Read up
  Read up

Function evaluate has 28 lines of code (exceeds 25 allowed). Consider refactoring.

Open

def evaluate(dataset_path: str, **kwargs):
    """
    Evaluate a trained model by predicting synthetic data and recording the image similarity

    :param dataset_path: Root path of the dataset containing videos of worm

Function _make_continuous_partitions has a Cognitive Complexity of 10 (exceeds 5 allowed). Consider refactoring.

Open

def _make_continuous_partitions(
    shuffled_results: ShuffledResults, score_threshold: float, frame_rate: float
) -> _PartitionedResults:
    time_window = max(1, int(frame_rate * CONTINOUS_SEGMENT_TIME_WINDOW_SEC))
    min_segment_size = max(1, int(frame_rate * MIN_SEGMENT_SIZE_SEC))

• Read up
  Read up

Function post_process has a Cognitive Complexity of 10 (exceeds 5 allowed). Consider refactoring.

Open

def post_process(dataset_path: str, **kwargs):
    """
    Process the raw network results with interpolation and smoothing

    :param dataset_path: Root path of the dataset containing videos of worm

• Read up
  Read up

Function save has a Cognitive Complexity of 10 (exceeds 5 allowed). Consider refactoring.

Open

    def save(self, results: dict, video_name: str, metadata: Optional[dict] = None):
        results_folder = os.path.join(self.results_root_dir, video_name)
        if not os.path.exists(results_folder):
            os.mkdir(results_folder)

• Read up
  Read up

Function generate has a Cognitive Complexity of 10 (exceeds 5 allowed). Consider refactoring.

Open

    def generate(
        self,
        theta: np.ndarray,
        template_frame: np.ndarray,
        template_skeleton: np.ndarray,

• Read up
  Read up

Function _write_to_file has 8 arguments (exceeds 4 allowed). Consider refactoring.

Open

def _write_to_file(

Function __init__ has 8 arguments (exceeds 4 allowed). Consider refactoring.

Open

    def __init__(

Function _align_segments_with_labels has a Cognitive Complexity of 9 (exceeds 5 allowed). Consider refactoring.

Open

def _align_segments_with_labels(segments, partitioned_skeletons, labelled_skeletons, min_labelled=5):
    """
    Match the head/tail alignment with the results of the classical tracking in each of the segments,
     if there is enough labelled data in the segment
    """

• Read up
  Read up

Function export has a Cognitive Complexity of 9 (exceeds 5 allowed). Consider refactoring.

Open

    def export(self, video_name: str, **kwargs):

        timestamp = kwargs["dataset"].features_dataset[video_name].timestamp
        results_skeletons: np.ndarray = kwargs["results_skeletons"]
        out_dir: str = kwargs["out_dir"]

• Read up
  Read up

Function _parse_arguments has a Cognitive Complexity of 9 (exceeds 5 allowed). Consider refactoring.

Open

def _parse_arguments(kwargs: dict):
    if kwargs.get("num_process") is None:
        kwargs["num_process"] = os.cpu_count()
    if kwargs.get("temp_dir") is None:
        kwargs["temp_dir"] = tempfile.gettempdir()

• Read up
  Read up

Function _parse_arguments has a Cognitive Complexity of 9 (exceeds 5 allowed). Consider refactoring.

Open

def _parse_arguments(dataset_path: str, kwargs: dict):
    if kwargs.get("work_dir") is None:
        kwargs["work_dir"] = default_paths.WORK_DIR
    if kwargs.get("max_gap_size") is None:
        kwargs["max_gap_size"] = 4

• Read up
  Read up

Function __init__ has 7 arguments (exceeds 4 allowed). Consider refactoring.

Open

    def __init__(

Function _validate_features has a Cognitive Complexity of 8 (exceeds 5 allowed). Consider refactoring.

Open

def _validate_features(raw_features: dict):
    if not (
        len(raw_features["skeletons"])
        == len(raw_features["head_width"])
        == len(raw_features["midbody_width"])

• Read up
  Read up

Function __init__ has 6 arguments (exceeds 4 allowed). Consider refactoring.

Open

    def __init__(

Function _align_unlabelled_segments_with_adjacents has a Cognitive Complexity of 8 (exceeds 5 allowed). Consider refactoring.

Open

def _align_unlabelled_segments_with_adjacents(segments, segments_alignment, partitioned_skeletons, frame_rate: float):
    """
    Resolve the unaligned segments by comparing with adjacent segments,
    starting with the segments that have the least frames gap between an adjacent trusted segment
    Don't align isolated segments which a big gap between trusted segments

• Read up
  Read up

Function generate has a Cognitive Complexity of 8 (exceeds 5 allowed). Consider refactoring.

Open

    def generate(self, file_pattern: str, num_samples: int):
        """
        Start the generation of files containing synthetic data

        :param file_pattern: Filepath of the files to generate with an "index" variable

• Read up
  Read up