Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
P
py-3d-face-reconstruction
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Analytics
Analytics
CI / CD
Repository
Value Stream
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Richard Torenvliet
py-3d-face-reconstruction
Commits
9712728e
Commit
9712728e
authored
8 years ago
by
Richard Torenvliet
Browse files
Options
Browse Files
Download
Plain Diff
Merge
parents
1895f423
5a5697bd
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
15 additions
and
21 deletions
+15
-21
src/server.py
src/server.py
+15
-20
src/settings.py
src/settings.py
+0
-1
No files found.
src/server.py
View file @
9712728e
...
...
@@ -25,6 +25,11 @@ DATASET = os.environ.get('DATASET', 'ibug') # see src/datasets for options
# load correct module to support the dataset
dataset_module
=
import_dataset_module
(
DATASET
)
images
=
glob
(
'{}/*.jpg'
.
format
(
FACE_DB
))
images
.
sort
()
asf
=
glob
(
'{}/*.asf'
.
format
(
FACE_DB
))
asf
.
sort
()
class
ImageWebSocketHandler
(
websocket
.
WebSocketHandler
):
handlers
=
{
...
...
@@ -33,12 +38,6 @@ class ImageWebSocketHandler(websocket.WebSocketHandler):
}
def
__init__
(
self
,
*
args
,
**
kwargs
):
self
.
images
=
glob
(
'{}/*.jpg'
.
format
(
FACE_DB
))
self
.
asf
=
glob
(
'{}/*.asf'
.
format
(
FACE_DB
))
self
.
images
.
sort
()
self
.
asf
.
sort
()
# todo get from settings
model_texture_file
=
'{}/pca_{}_texture_model.npy'
.
format
(
FILES_DIR
,
DATASET
)
...
...
@@ -81,9 +80,9 @@ class ImageWebSocketHandler(websocket.WebSocketHandler):
t1
=
time
()
if
DATASET
==
'imm'
:
image_filename
=
self
.
asf
[
image_index
]
image_filename
=
asf
[
image_index
]
else
:
image_filename
=
self
.
images
[
image_index
]
image_filename
=
images
[
image_index
]
dst_image
=
reconstruction
.
reconstruct_shape_texture
(
dataset_module
,
...
...
@@ -114,15 +113,15 @@ class ImageWebSocketHandler(websocket.WebSocketHandler):
for
m
in
message
.
keys
():
try
:
handler
=
getattr
(
self
,
self
.
handlers
[
m
])
print
handler
logger
.
debug
(
'calling handler %s'
,
handler
)
handler
(
message
[
m
])
except
(
AttributeError
,
KeyError
)
as
e
:
msg
=
'no handler for {}
'
.
format
(
m
)
print
(
msg
,
e
)
msg
=
'no handler for {}
: {}'
.
format
(
m
,
e
)
logger
.
info
(
'%s'
,
msg
)
self
.
__return_error
(
msg
)
except
Exception
as
e
:
msg
=
'no handler for {}
'
.
format
(
m
)
print
(
msg
,
e
)
msg
=
'no handler for {}
: {}'
.
format
(
m
,
e
)
logger
.
info
(
'%s'
,
msg
)
self
.
__return_error
(
msg
)
traceback
.
print_exc
()
...
...
@@ -132,11 +131,6 @@ class ImageWebSocketHandler(websocket.WebSocketHandler):
class
ApiHandler
(
web
.
RequestHandler
):
def
__init__
(
self
,
*
args
,
**
kwargs
):
self
.
images
=
glob
(
'{}/*.jpg'
.
format
(
FACE_DB
))
self
.
asf_files
=
glob
(
'{}/*.asf'
.
format
(
FACE_DB
))
self
.
images
.
sort
()
self
.
asf_files
.
sort
()
web
.
RequestHandler
.
__init__
(
self
,
*
args
,
**
kwargs
)
def
set_default_headers
(
self
):
...
...
@@ -145,6 +139,7 @@ class ApiHandler(web.RequestHandler):
self
.
set_header
(
'Access-Control-Allow-Methods'
,
'POST, GET, OPTIONS'
)
self
.
set_header
(
"Content-Type"
,
'application/vnd.api+json'
)
class
FaceHandler
(
ApiHandler
):
@
web
.
asynchronous
def
get
(
self
,
*
args
):
...
...
@@ -154,13 +149,13 @@ class FaceHandler(ApiHandler):
"""
data
=
[]
for
id
,
filename
in
enumerate
(
self
.
asf_files
):
for
id
,
filename
in
enumerate
(
asf
):
data
.
append
({
'type'
:
'faces'
,
'id'
:
id
,
'attributes'
:
{
'filename'
:
'{}/{}'
.
format
(
FACE_DB_NAME
,
os
.
path
.
basename
(
self
.
images
[
id
])
FACE_DB_NAME
,
os
.
path
.
basename
(
images
[
id
])
),
}
})
...
...
This diff is collapsed.
Click to expand it.
src/settings.py
View file @
9712728e
...
...
@@ -48,7 +48,6 @@ logging.config.dictConfig({
})
logger
=
logging
.
getLogger
(
'root'
)
#logger.setLevel(logging.DEBUG)
if
os
.
environ
.
get
(
'DEBUG'
,
False
):
logger
=
logging
.
getLogger
(
'debug'
)
This diff is collapsed.
Click to expand it.
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