Coder Social home page Coder Social logo

nbnet's Introduction

Code for Paper:

G. Mai, Kai Cao, P. C. Yuen and Anil K. Jain, "On the Reconstruction of Face Images from Deep Face Templates", 
IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI), 2019

Please cite our paper if you find the code useful. 

Clone the project from the github with 

git clone [email protected]:csgcmai/NBNet.git

Install the required packages, the usage of the conda enviornment is recommended.
Run the following commands

conda create -n nbnet python=2.7 anaconda
pip install --upgrade pip

Piror to the next step, please refer https://www.tensorflow.org/install/gpu 
and http://mxnet.io/install/index.html for installing the necessary packages 
for installing the tensorflow and mxnet with GPU support


cd NBNet
pip install -r requirement.txt

cd src
python train_of2img_mae.py --gpus 0,1,2,3

nbnet's People

Contributors

csgcmai avatar

Stargazers

avatar avatar avatar avatar avatar avatar avatar avatar avatar avatar

Watchers

avatar avatar avatar

Forkers

courageface alwaysnow magicalfish zhangzp9970

nbnet's Issues

RuntimeError: simple_bind error. Arguments: data: (64, 100)

**Hi,

I am trying to run the "train_of2img_mae.py" code on Colab using mxnet_cu11.1 and CUDA 11.1.
I get the following error message: (When I try it using mxnet-cu10.1 and CUDA 10.1, the same error occurs)
Can you please advise me regarding this error?**

`WARNING:root:Limited tf.compat.v2.summary API due to missing TensorBoard installation.
WARNING:root:Limited tf.compat.v2.summary API due to missing TensorBoard installation.
WARNING:root:Limited tf.compat.v2.summary API due to missing TensorBoard installation.
WARNING:root:Limited tf.summary API due to missing TensorBoard installation.
WARNING:root:Limited tf.compat.v2.summary API due to missing TensorBoard installation.
WARNING:root:Limited tf.compat.v2.summary API due to missing TensorBoard installation.
WARNING:root:Limited tf.compat.v2.summary API due to missing TensorBoard installation.
/usr/local/lib/python3.7/dist-packages/mxnet/module/base_module.py:55: UserWarning: You created Module with Module(..., label_names=['softmax_label']) but input with name 'softmax_label' is not found in symbol.list_arguments(). Did you mean one of:
data
warnings.warn(msg)
/usr/local/lib/python3.7/dist-packages/mxnet/module/base_module.py:67: UserWarning: Data provided by label_shapes don't match names specified by label_names ([] vs. ['softmax_label'])
warnings.warn(msg)
Traceback (most recent call last):
File "/usr/local/lib/python3.7/dist-packages/mxnet/symbol/symbol.py", line 1938, in simple_bind
ctypes.byref(exe_handle)))
File "/usr/local/lib/python3.7/dist-packages/mxnet/base.py", line 246, in check_call
raise get_last_ffi_error()
mxnet.base.MXNetError: Traceback (most recent call last):
File "../src/engine/./../common/cuda_utils.h", line 394
CUDA: Check failed: e == cudaSuccess || e == cudaErrorCudartUnloading: invalid device ordinal

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "train_of2img_mae.py", line 159, in
main(args)
File "train_of2img_mae.py", line 64, in main
model_prefix=args.face_gen_prefix, load_epoch=args.face_gen_epoch)
File "/content/drive/MyDrive/NBNet/src/util/util.py", line 63, in img_generator
model_trained.bind(data_shapes=[('data',(batch_size,100))],label_shapes=None, for_training=False)
File "/usr/local/lib/python3.7/dist-packages/mxnet/module/module.py", line 429, in bind
state_names=self._state_names)
File "/usr/local/lib/python3.7/dist-packages/mxnet/module/executor_group.py", line 280, in init
self.bind_exec(data_shapes, label_shapes, shared_group)
File "/usr/local/lib/python3.7/dist-packages/mxnet/module/executor_group.py", line 384, in bind_exec
shared_group))
File "/usr/local/lib/python3.7/dist-packages/mxnet/module/executor_group.py", line 678, in _bind_ith_exec
shared_buffer=shared_data_arrays, **input_shapes)
File "/usr/local/lib/python3.7/dist-packages/mxnet/symbol/symbol.py", line 1944, in simple_bind
raise RuntimeError(error_msg)
RuntimeError: simple_bind error. Arguments:
data: (64, 100)
Traceback (most recent call last):
File "../src/engine/./../common/cuda_utils.h", line 394
CUDA: Check failed: e == cudaSuccess || e == cudaErrorCudartUnloading: invalid device ordinal`

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.