search

deepinsight / insightface

State-of-the-art 2D and 3D Face Analysis Project
https://insightface.ai
22.74k stars 5.33k forks source link

When using Python 3, I found a very strange problem. Please let me know. Thank you very much #1103

Open xianglei3 opened 4 years ago

xianglei3 commented 4 years ago

call reset() Traceback (most recent call last): File "train.py", line 1025, in main() File "train.py", line 1022, in main train_net(args) File "train.py", line 1016, in train_net epoch_end_callback = epoch_cb ) File "/home/xl/anaconda3/envs/test/lib/python3.5/site-packages/mxnet/module/base_module.py", line 520, in fit next_data_batch = next(data_iter) File "/home/xl/anaconda3/envs/test/lib/python3.5/site-packages/mxnet/io/io.py", line 230, in next return self.next() File "/home/xl/1mxnetinsight/insightface-master/src/data.py", line 860, in next self.reset() File "/home/xl/1mxnetinsight/insightface-master/src/data.py", line 753, in reset random.shuffle(self.seq) File "/home/xl/anaconda3/envs/test/lib/python3.5/random.py", line 281, in shuffle x[i], x[j] = x[j], x[i] TypeError: 'range' object does not support item assignment (test) xl@xl-Alienware-13-R3:~/1mxnetinsight/insightface-master/src$

tbuikr commented 4 years ago

In python3, the range does not return a list, so you should convert self.seq to list in line 753 of data.py

random.shuffle(list(self.seq))