xuehy/pytorch-gaitgan

GaitGAN: Invariant Gait Feature Extraction Using Generative Adversarial Networks

pytorch-gaitgan's Issues

generate.py can not generate the image which angle is 180 because the number of all angle is 11 but the line #41 of generate.py use the range(10)

CASIA-B dataset is not match the dataset.py

Hi, the dataset.py is not fit the CASIA-B dataset, how we should process the dataset and then run the train.py?

Why is it stuck at training starts when I run python3 train.py

Why is it stuck at training starts when I run python3 train.py，and iteration keeps 0

How to preprocess the silhouette image to GEI

Hi, can I know more detail about how did you preprocess the silhouette image to GEI?
I have run the code successfully, but all my GEI(the following picture) aren't as clear as yours.
thank you very much!!

Cuda Error while training

Exception in user code:

Traceback (most recent call last):
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/urllib3/connection.py", line 171, in _new_conn
(self._dns_host, self.port), self.timeout, **extra_kw)
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/urllib3/util/connection.py", line 79, in create_connection
raise err
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/urllib3/util/connection.py", line 69, in create_connection
sock.connect(sa)
ConnectionRefusedError: [Errno 111] Connection refused

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/urllib3/connectionpool.py", line 600, in urlopen
chunked=chunked)
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/urllib3/connectionpool.py", line 354, in _make_request
conn.request(method, url, **httplib_request_kw)
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/http/client.py", line 1239, in request
self._send_request(method, url, body, headers, encode_chunked)
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/http/client.py", line 1285, in _send_request
self.endheaders(body, encode_chunked=encode_chunked)
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/http/client.py", line 1234, in endheaders
self._send_output(message_body, encode_chunked=encode_chunked)
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/http/client.py", line 1026, in _send_output
self.send(msg)
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/http/client.py", line 964, in send
self.connect()
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/urllib3/connection.py", line 196, in connect
conn = self._new_conn()
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/urllib3/connection.py", line 180, in _new_conn
self, "Failed to establish a new connection: %s" % e)
urllib3.exceptions.NewConnectionError: <urllib3.connection.HTTPConnection object at 0x7faa21e4fcf8>: Failed to establish a new connection: [Errno 111] Connection refused

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/requests/adapters.py", line 445, in send
timeout=timeout
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/urllib3/connectionpool.py", line 638, in urlopen
_stacktrace=sys.exc_info()[2])
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/urllib3/util/retry.py", line 398, in increment
raise MaxRetryError(_pool, url, error or ResponseError(cause))
urllib3.exceptions.MaxRetryError: HTTPConnectionPool(host='localhost', port=5274): Max retries exceeded with url: /env/main (Caused by NewConnectionError('<urllib3.connection.HTTPConnection object at 0x7faa21e4fcf8>: Failed to establish a new connection: [Errno 111] Connection refused',))

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/visdom/init.py", line 446, in _send
data=json.dumps(msg),
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/requests/api.py", line 112, in post
return request('post', url, data=data, json=json, **kwargs)
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/requests/api.py", line 58, in request
return session.request(method=method, url=url, **kwargs)
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/requests/sessions.py", line 512, in request
resp = self.send(prep, **send_kwargs)
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/requests/sessions.py", line 622, in send
r = adapter.send(request, **kwargs)
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/requests/adapters.py", line 513, in send
raise ConnectionError(e, request=request)
requests.exceptions.ConnectionError: HTTPConnectionPool(host='localhost', port=5274): Max retries exceeded with url: /env/main (Caused by NewConnectionError('<urllib3.connection.HTTPConnection object at 0x7faa21e4fcf8>: Failed to establish a new connection: [Errno 111] Connection refused',))
Traceback (most recent call last):
File "train.py", line 37, in
netg = netg.to(device)
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/torch/nn/modules/module.py", line 393, in to
return self._apply(lambda t: t.to(device))
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/torch/nn/modules/module.py", line 176, in _apply
module._apply(fn)
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/torch/nn/modules/module.py", line 176, in _apply
module._apply(fn)
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/torch/nn/modules/module.py", line 182, in _apply
param.data = fn(param.data)
File "/home/administrator/anaconda2/envs/deeplearning/lib/python3.6/site-packages/torch/nn/modules/module.py", line 393, in
return self._apply(lambda t: t.to(device))
RuntimeError: CUDA error (10): invalid device ordinal

Code Consulting

I want to research GaitGAN code, so what is your detailed configuration environment?