Module ImportFrom alias ImportFrom alias Import alias identifier Assign NameStore ListLoad Str Str ClassDef bases NameLoad body Expr Str FunctionDef arguments args NameParam NameParam NameParam defaults Num body Assert CompareEq Call NameLoad NameLoad Num Assign AttributeStore NameLoad attr NameLoad Assign AttributeStore NameLoad attr NameLoad decorator_list FunctionDef arguments args NameParam NameParam defaults body Assign NameStore Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr keyword Str Assign TupleStore NameStore NameStore SubscriptLoad AttributeLoad NameLoad attr ExtSlice Slice SubscriptLoad AttributeLoad NameLoad attr Index Num Slice SubscriptLoad AttributeLoad NameLoad attr Index Num Assign NameStore BinOpSub BinOpDiv Call AttributeLoad NameLoad attr Str SubscriptLoad AttributeLoad NameLoad attr Index Num SubscriptLoad NameLoad Index Num Assign NameStore BinOpSub BinOpDiv Call AttributeLoad NameLoad attr Str SubscriptLoad AttributeLoad NameLoad attr Index Num SubscriptLoad NameLoad Index Num Assign NameStore Call AttributeLoad NameLoad attr BinOpDiv UnaryOpUSub BinOpAdd BinOpPow NameLoad Num BinOpPow NameLoad Num AttributeLoad NameLoad attr Return NameLoad decorator_list decorator_list FunctionDef arguments args NameParam NameParam defaults body Assign NameStore Call AttributeLoad NameLoad attr NameLoad Num Assign NameStore Call AttributeLoad NameLoad attr NameLoad Call AttributeLoad NameLoad attr ListLoad BinOpSub SubscriptLoad AttributeLoad NameLoad attr Index Num Num BinOpSub SubscriptLoad AttributeLoad NameLoad attr Index Num Num Assign NameStore Call AttributeLoad Call AttributeLoad NameLoad attr NameLoad attr Str Assign NameStore BinOpAdd NameLoad Num Assign NameStore Call AttributeLoad NameLoad attr NameLoad Call AttributeLoad NameLoad attr ListLoad BinOpSub SubscriptLoad AttributeLoad NameLoad attr Index Num Num BinOpSub SubscriptLoad AttributeLoad NameLoad attr Index Num Num Assign NameStore BinOpSub NameLoad NameLoad Assign NameStore BinOpSub Num NameLoad Assign TupleStore NameStore NameStore Call AttributeLoad NameLoad attr NameLoad Num keyword Num Assign TupleStore NameStore NameStore Call AttributeLoad NameLoad attr NameLoad Num keyword Num Assign NameStore Call AttributeLoad Call AttributeLoad NameLoad attr NameLoad Num Num attr TupleLoad SubscriptLoad AttributeLoad NameLoad attr Index Num SubscriptLoad AttributeLoad NameLoad attr Index Num Num Num Assign NameStore Call AttributeLoad Call AttributeLoad NameLoad attr NameLoad Num Num attr TupleLoad SubscriptLoad AttributeLoad NameLoad attr Index Num SubscriptLoad AttributeLoad NameLoad attr Index Num Num Num Assign TupleStore NameStore NameStore Call AttributeLoad NameLoad attr NameLoad Num keyword Num Assign TupleStore NameStore NameStore Call AttributeLoad NameLoad attr NameLoad Num keyword Num Assign NameStore BinOpAdd BinOpAdd BinOpAdd BinOpMult BinOpMult SubscriptLoad NameLoad ExtSlice Index NameLoad Index NameLoad Slice NameLoad NameLoad BinOpMult BinOpMult SubscriptLoad NameLoad ExtSlice Index NameLoad Index NameLoad Slice NameLoad NameLoad BinOpMult BinOpMult SubscriptLoad NameLoad ExtSlice Index NameLoad Index NameLoad Slice NameLoad NameLoad BinOpMult BinOpMult SubscriptLoad NameLoad ExtSlice Index NameLoad Index NameLoad Slice NameLoad NameLoad Return SubscriptLoad NameLoad ExtSlice Slice Slice Index Num Slice decorator_list ClassDef bases NameLoad body Expr Str FunctionDef arguments args NameParam NameParam NameParam NameParam NameParam defaults Num NameLoad body Expr Str Expr Call AttributeLoad NameLoad attr Str Expr Call AttributeLoad Call NameLoad NameLoad NameLoad attr Assign AttributeStore NameLoad attr NameLoad Assign AttributeStore NameLoad attr Call NameLoad AttributeLoad NameLoad attr Assign AttributeStore NameLoad attr NameLoad Assign AttributeStore NameLoad attr Call AttributeLoad SubscriptLoad AttributeLoad NameLoad attr ExtSlice Slice Num SubscriptLoad AttributeLoad NameLoad attr Index Num Slice Num SubscriptLoad AttributeLoad NameLoad attr Index Num attr Num Num Num Assign AttributeStore NameLoad attr Call AttributeLoad AttributeLoad NameLoad attr attr Str Assign NameStore Call NameLoad AttributeLoad NameLoad attr keyword NameLoad Assign AttributeStore NameLoad attr Call AttributeLoad NameLoad attr ListComp Call AttributeLoad NameLoad attr NameLoad comprehension NameStore AttributeLoad NameLoad attr keyword Str Assign AttributeStore NameLoad attr Call AttributeLoad AttributeLoad NameLoad attr attr Num Num Num If CompareIs NameLoad NameLoad body Assign AttributeStore NameLoad attr BinOpDiv SubscriptLoad AttributeLoad NameLoad attr Index Num Num orelse Assign AttributeStore NameLoad attr NameLoad decorator_list FunctionDef arguments args NameParam NameParam defaults body If AttributeLoad NameLoad attr body Raise Call NameLoad Assign NameStore BinOpSub Call AttributeLoad Call AttributeLoad AttributeLoad NameLoad attr attr AttributeLoad NameLoad attr Num attr Str Num Assign NameStore BinOpMult BinOpMult NameLoad Num AttributeLoad NameLoad attr Assign NameStore BinOpAdd AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr NameLoad Assign AttributeStore NameLoad attr Call NameLoad AttributeLoad NameLoad attr NameLoad decorator_list decorator_list
N/A base ImageAugmentor utils logger N/A numpy np N/A __all__ N/A GaussianDeform GaussianMap GaussianMap N/A object N/A N/A  Generate gaussian weighted deformation map __init__ N/A N/A self image_shape sigma N/A 0.5 N/A N/A N/A N/A len image_shape 2 N/A N/A self shape image_shape N/A N/A self sigma sigma N/A get_gaussian_weight N/A N/A self anchor N/A N/A N/A ret N/A N/A np zeros N/A self shape dtype float32 N/A N/A y x N/A N/A np mgrid N/A N/A N/A N/A self shape N/A 0 N/A N/A N/A self shape N/A 1 N/A y N/A N/A N/A N/A y astype float32 N/A N/A ret shape N/A 0 N/A anchor N/A 0 N/A x N/A N/A N/A N/A x astype float32 N/A N/A ret shape N/A 1 N/A anchor N/A 1 N/A g N/A N/A np exp N/A N/A N/A N/A x 2 N/A y 2 N/A self sigma N/A g N/A N/A np_sample N/A N/A img coords N/A N/A N/A coords N/A N/A np maximum coords 0 N/A coords N/A N/A np minimum coords N/A N/A np array N/A N/A N/A N/A img shape N/A 0 1 N/A N/A N/A img shape N/A 1 1 N/A lcoor N/A N/A N/A N/A np floor coords astype int32 N/A ucoor N/A lcoor 1 N/A ucoor N/A N/A np minimum ucoor N/A N/A np array N/A N/A N/A N/A img shape N/A 0 1 N/A N/A N/A img shape N/A 1 1 N/A diff N/A coords lcoor N/A neg_diff N/A 1.0 diff N/A N/A lcoory lcoorx N/A N/A np split lcoor 2 axis 2 N/A N/A ucoory ucoorx N/A N/A np split ucoor 2 axis 2 N/A diff N/A N/A N/A N/A np repeat diff 3 2 reshape N/A N/A N/A diff shape N/A 0 N/A N/A diff shape N/A 1 2 3 N/A neg_diff N/A N/A N/A N/A np repeat neg_diff 3 2 reshape N/A N/A N/A diff shape N/A 0 N/A N/A diff shape N/A 1 2 3 N/A N/A diffy diffx N/A N/A np split diff 2 axis 2 N/A N/A ndiffy ndiffx N/A N/A np split neg_diff 2 axis 2 N/A ret N/A N/A N/A N/A N/A N/A img N/A N/A lcoory N/A lcoorx N/A ndiffx ndiffy N/A N/A N/A img N/A N/A ucoory N/A ucoorx N/A diffx diffy N/A N/A N/A img N/A N/A lcoory N/A ucoorx N/A ndiffy diffx N/A N/A N/A img N/A N/A ucoory N/A lcoorx N/A diffy ndiffx N/A N/A ret N/A N/A N/A N/A 0 N/A N/A GaussianDeform N/A ImageAugmentor N/A N/A 
    Some kind of deformation. Quite slow.
     __init__ N/A N/A self anchors shape sigma randrange N/A 0.5 None N/A N/A 
        :param anchors: in [0,1] coordinate
        :param shape: image shape in [h, w]
        :param sigma: sigma for Gaussian weight
        :param randrange: default to shape[0] / 8
         N/A N/A N/A logger warn GaussianDeform is slow. Consider using it with 4 or more prefetching processes. N/A N/A N/A N/A super GaussianDeform self __init__ N/A N/A self anchors anchors N/A N/A self K N/A len N/A self anchors N/A N/A self shape shape N/A N/A self grid N/A N/A N/A N/A np mgrid N/A N/A 0 N/A N/A self shape N/A 0 N/A 0 N/A N/A self shape N/A 1 transpose 1 2 0 N/A N/A self grid N/A N/A N/A self grid astype float32 N/A gm N/A GaussianMap N/A self shape sigma sigma N/A N/A self gws N/A N/A np array N/A N/A N/A gm get_gaussian_weight ank N/A ank N/A self anchors dtype float32 N/A N/A self gws N/A N/A N/A self gws transpose 1 2 0 N/A N/A randrange None N/A N/A N/A self randrange N/A N/A N/A self shape N/A 0 8 N/A N/A N/A self randrange randrange N/A _augment N/A N/A self img N/A N/A N/A N/A img coords N/A N/A N/A NotImplementedError N/A v N/A N/A N/A N/A N/A N/A self rng rand N/A self K 2 astype float32 0.5 N/A v N/A N/A v 2 N/A self randrange N/A grid N/A N/A self grid N/A N/A np dot N/A self gws v N/A N/A img arr N/A np_sample N/A img arr grid N/A N/A