NLeSC / PattyAnalytics

scripts/redstickdetection.py

#!/usr/bin/env python
"""Segment points by colour from a pointcloud file and saves all reddish points
target pointcloud file. Autodectects ply, pcd and las files.

Usage: redstickdetection.py  [-h] <infile> <outfile>
"""

from docopt import docopt
from patty.segmentation.segRedStick import get_red_mask
from patty.utils import extract_mask, load, save

if __name__ == '__main__':
    args = docopt(__doc__)

The name args does not conform to the constant naming conventions ((([A-Z_][A-Z0-9_]*)|(__.*__))$).

    pc = load(args['<infile>'])

The name pc does not conform to the constant naming conventions ((([A-Z_][A-Z0-9_]*)|(__.*__))$).

    red_pc = extract_mask(pc, get_red_mask(pc))

The name red_pc does not conform to the constant naming conventions ((([A-Z_][A-Z0-9_]*)|(__.*__))$).

    save(red_pc, args['<outfile>'])