import numpy as np
import pandas as pd
from pandas import DataFrame, Series  # for convenience

import pims
import trackpy as tp

import os
import sys


@pims.pipeline
def gray(image):
    return image[:, :, 1]  # Take just the green channel


if __name__=="__main__":

    frames = gray(pims.open(os.path.join(sys.argv[1], '*.bmp')))

    print(frames)

    f = tp.locate(frames[0], diameter=5, minmass=20, maxsize=10, invert=False)

    print(f.head())

    tp.annotate(f, frames[0])