ven-kyoshiro

import matplotlib.pyplot as plt
import numpy as np

def val2color(x):
    assert x<=1, 'x should be in the [0,1]'
    base =0.5
    if x<1/3:
        g=x*3.
        r= 1-g
        b=0

ven-kyoshiro

import tqdm
import time
import pandas as pd
import numpy as np
import cv2
import matplotlib.pyplot as plt

from sklearn.cluster import DBSCAN

def measure_position(arr,mouse_area_max,mouse_area_min,

ven-kyoshiro

"""
# 変更点
- video2activityで画像を出力するフォルダを指定できるように
  - video2activity('200922_4_10min.mp4','test.csv',visualize=True,visualize_dir='hoge')
  - ↑の場合 hoge以下に画像が出力される
- 動画内でずっと黒色になっている部分を検出して，白く塗りつぶす処理を追加
  - video2activity('200922_4_10min.mp4','test.csv',remove_dontmove=True)
"""
import tqdm
import time

ven-kyoshiro

def bb_intersection_over_union(boxA, boxB):
    """ calculate IOU score between boxA and boxB
    Args:
        boxA : [hori_lo,virt_lo,hori_hi,virt_hi]
        boxB : [hori_lo,virt_lo,hori_hi,virt_hi]
    Returns:
        iou_score
    """
    # determine the (x, y)-coordinates of the intersection rectangle
    xA = max(boxA[0], boxB[0])

ven-kyoshiro

import matplotlib.pyplot as plt
import matplotlib.animation as animation
import numpy as np
"""
marge two scatter animation


"""

fig = plt.figure()

ven-kyoshiro

import numpy as np
def ccw(A,B,C):
    return (C[1]-A[1]) * (B[0]-A[0]) > (B[1]-A[1]) * (C[0]-A[0])

def intersect(A,B,C,D):
    # check AB vs CD
    return ccw(A,C,D) != ccw(B,C,D) and ccw(A,B,C) != ccw(A,B,D)

class LineSegment:
    def __init__(self,xy1,xy2):

ven-kyoshiro

import numpy as np
import matplotlib.pyplot as plt
def ccw(A,B,C):
    return (C[1]-A[1]) * (B[0]-A[0]) > (B[1]-A[1]) * (C[0]-A[0])

def intersect(A,B,C,D):
    # check AB vs CD
    return ccw(A,C,D) != ccw(B,C,D) and ccw(A,B,C) != ccw(A,B,D)

class LineSegment: