-
Notifications
You must be signed in to change notification settings - Fork 68
Description
Hello,
I try to execute your code but there is problem, I cant find any solution
Can you please help me.
I download the dataset wider face as you explain and I tried to run this command on windows:
set CUDA_VISIBLE_DEVICES=0 & python train.py --data_path dataset/widerface --batch 1 --save_path ./out
but I get this problem:
Namespace(batch=1, data_path='dataset/widerface', depth=50, epochs=1, eval_step=3, img_size=512, save_path='./out', save_step=10, shuffle=True, verbose=10)
Traceback (most recent call last):
File "train.py", line 151, in
main()
File "train.py", line 55, in main
dataset_val = ValDataset(val_path,transform=transforms.Compose([RandomCroper()]))
File "C:\Desktop\RetinaFace_super\dataloader.py", line 332, in init
label = [float(x) for x in line]
File "C:\Desktop\RetinaFace_super\dataloader.py", line 332, in
label = [float(x) for x in line]
ValueError: could not convert string to float: '/24--Soldier_Firing/24_Soldier_Firing_Soldier_Firing_24_329.jpg'
when I change the val images with the same as train images it start the training then I get this error :
---- [Epoch 0/1, Batch 12870/12880] ----
+----------------+---------------------+
| loss name | value |
+----------------+---------------------+
| total_loss | 2.6635076999664307 |
| classification | 1.5447975397109985 |
| bbox | 0.34370726346969604 |
| landmarks | 0.7750030159950256 |
+----------------+---------------------+
-------- RetinaFace Pytorch --------
Evaluating epoch 0
0%| | 0/12880 [00:00<?, ?it/s]Traceback (most recent call last):
File "train.py", line 151, in
main()
File "train.py", line 136, in main
recall, precision = eval_widerface.evaluate(dataloader_val,retinaface)
File "C:\Desktop\RetinaFace_super\eval_widerface.py", line 74, in evaluate
for data in tqdm(iter(val_data)):
File "C:\AppData\Local\Continuum\anaconda3\envs\supernotman\lib\site-packages\tqdm\std.py", line 1099, in iter
for obj in iterable:
File "C:\AppData\Local\Continuum\anaconda3\envs\supernotman\lib\site-packages\torch\utils\data\dataloader.py", line 582, in next
return self._process_next_batch(batch)
File "C:\AppData\Local\Continuum\anaconda3\envs\supernotman\lib\site-packages\torch\utils\data\dataloader.py", line 608, in _process_next_batch
raise batch.exc_type(batch.exc_msg)
ValueError: Traceback (most recent call last):
File "C:\AppData\Local\Continuum\anaconda3\envs\supernotman\lib\site-packages\torch\utils\data_utils\worker.py", line 99, in _worker_loop
samples = collate_fn([dataset[i] for i in batch_indices])
File "C:\AppData\Local\Continuum\anaconda3\envs\supernotman\lib\site-packages\torch\utils\data_utils\worker.py", line 99, in
samples = collate_fn([dataset[i] for i in batch_indices])
File "C:\Desktop\RetinaFace_super\dataloader.py", line 347, in getitem
annotation[0,0] = label[0] # x1
ValueError: setting an array element with a sequence.
0%|
I dont know what s go on
I really appreciate if you help me.