Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Lucas Miranda
deepOF
Commits
fdf2ddab
Commit
fdf2ddab
authored
Mar 26, 2021
by
lucas_miranda
Browse files
Updated DockerFile to use a specific tag in the FROM command
parent
f05b62df
Pipeline
#96944
passed with stages
in 19 minutes and 6 seconds
Changes
3
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
deepof/model_utils.py
View file @
fdf2ddab
...
...
@@ -287,7 +287,9 @@ class neighbor_latent_entropy(tf.keras.callbacks.Callback):
with
writer
.
as_default
():
tf
.
summary
.
scalar
(
"average_neighborhood_cluster_entropy"
,
data
=
np
.
average
(
purity_vector
),
#, weights=max_groups[random_idxs]),
data
=
np
.
average
(
purity_vector
),
# , weights=max_groups[random_idxs]),
step
=
epoch
,
)
tf
.
summary
.
scalar
(
...
...
deepof/utils.py
View file @
fdf2ddab
...
...
@@ -556,13 +556,13 @@ def circular_arena_recognition(frame: np.array) -> np.array:
# Convert image to greyscale, threshold it, blur it and apply open-close operations
gray_image
=
cv2
.
cvtColor
(
frame
,
cv2
.
COLOR_BGR2GRAY
)
ret
,
thresh
=
cv2
.
threshold
(
gray_image
,
90
,
255
,
0
)
ret
,
thresh
=
cv2
.
threshold
(
gray_image
,
255
//
2
,
255
,
0
)
frame
=
cv2
.
medianBlur
(
thresh
,
9
)
frame
=
cv2
.
morphologyEx
(
frame
,
cv2
.
MORPH_CLOSE
,
cv2
.
getStructuringElement
(
cv2
.
MORPH_ELLIPSE
,
(
8
0
,
8
0
))
frame
,
cv2
.
MORPH_CLOSE
,
cv2
.
getStructuringElement
(
cv2
.
MORPH_ELLIPSE
,
(
5
0
,
5
0
))
)
frame
=
cv2
.
morphologyEx
(
frame
,
cv2
.
MORPH_OPEN
,
cv2
.
getStructuringElement
(
cv2
.
MORPH_ELLIPSE
,
(
8
0
,
8
0
))
frame
,
cv2
.
MORPH_OPEN
,
cv2
.
getStructuringElement
(
cv2
.
MORPH_ELLIPSE
,
(
5
0
,
5
0
))
)
# Find contours in the processed image
...
...
tests/test_pose_utils.py
View file @
fdf2ddab
...
...
@@ -121,7 +121,7 @@ def test_climb_wall(center, axes, angle, tol):
climb_wall
(
""
,
arena
,
prun
[
"test"
],
tol1
,
nose
=
"Nose"
)
@
settings
(
deadline
=
None
)
@
settings
(
deadline
=
None
,
suppress_health_check
=
[
HealthCheck
.
too_slow
]
)
@
given
(
pos_dframe
=
data_frames
(
index
=
range_indexes
(
min_size
=
5
),
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment