chenjun2hao / CenterFace.pytorch

unofficial version of centerface, which achieves the best balance between speed and accuracy at face detection

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

关于数据集的报错,没有找到train的dataset。

xggyh opened this issue · comments

Fix size testing.
training chunk_sizes: [15]
The output will be saved to /content/drive/My Drive/CenterFace.pytorch/src/lib/../../exp/multi_pose/mobilev2_10
heads {'hm': 1, 'wh': 2, 'hm_offset': 2, 'landmarks': 10}
Namespace(K=200, aggr_weight=0.0, agnostic_ex=False, arch='mobilev2_10', aug_ddd=0.5, aug_rot=0, batch_size=8, cat_spec_wh=False, center_thresh=0.1, chunk_sizes=[15], data_dir='/content/drive/My Drive/CenterFace.pytorch/src/lib/../../data', dataset='facehp', debug=0, debug_dir='/content/drive/My Drive/CenterFace.pytorch/src/lib/../../exp/multi_pose/mobilev2_10/debug', debugger_theme='white', demo='/home/yangna/data/WIDER_FACE/WIDER_train/images/0--Parade/0_Parade_marchingband_1_80.jpg', dense_hp=False, dense_wh=False, dep_weight=1, dim_weight=1, down_ratio=4, eval_oracle_dep=False, eval_oracle_hm=False, eval_oracle_hmhp=False, eval_oracle_hp_offset=False, eval_oracle_kps=False, eval_oracle_offset=False, eval_oracle_wh=False, exp_dir='/content/drive/My Drive/CenterFace.pytorch/src/lib/../../exp/multi_pose', exp_id='dla', fix_res=True, flip=0.5, flip_idx=[[0, 1], [3, 4]], flip_test=False, gpus=[0], gpus_str='0', head_conv=64, heads={'hm': 1, 'wh': 2, 'hm_offset': 2, 'landmarks': 10}, hide_data_time=False, hm_hp=True, hm_hp_weight=1, hm_weight=1, input_h=800, input_res=800, input_w=800, keep_res=False, kitti_split='3dop', lm_weight=0.1, load_model='', lr=0.000125, lr_step=[30, 80], master_batch_size=15, mean=array([[[0.40789655, 0.44719303, 0.47026116]]], dtype=float32), metric='loss', mse_loss=False, nms=False, no_color_aug=False, norm_wh=False, not_cuda_benchmark=False, not_hm_hp=False, not_prefetch_test=False, not_rand_crop=False, not_reg_bbox=False, not_reg_hp_offset=False, not_reg_offset=False, num_classes=1, num_epochs=140, num_iters=-1, num_stacks=1, num_workers=4, off_weight=1, output_h=200, output_res=200, output_video='../output/res_3.mp4', output_w=200, pad=31, peak_thresh=0.2, print_iter=0, rect_mask=False, reg_bbox=True, reg_hp_offset=True, reg_loss='sl1', reg_offset=True, resume=False, root_dir='/content/drive/My Drive/CenterFace.pytorch/src/lib/../..', rot_weight=1, rotate=0, save_all=True, save_dir='/content/drive/My Drive/CenterFace.pytorch/src/lib/../../exp/multi_pose/mobilev2_10', scale=0.4, scores_thresh=0.1, seed=317, shift=0.1, std=array([[[0.2886383 , 0.27408165, 0.27809834]]], dtype=float32), task='multi_pose', test=False, test_scales=[1.0], train_json=None, trainval=False, val_intervals=5, val_json=None, vis_thresh=0.4, wh_weight=0.1)
cp: target 'Drive/CenterFace.pytorch/src/lib/../../exp/multi_pose/mobilev2_10/logs_2020-08-19-12-45/' is not a directory
Creating model...
Setting up data...
==> initializing centerface key point val data.
loading annotations into memory...
Done (t=0.00s)
creating index...
index created!
Loaded val 0 samples
==> initializing centerface key point train data.
loading annotations into memory...
Done (t=0.77s)
creating index...
index created!
Loaded train 12671 samples
Starting training...
multi_pose/dlaTraceback (most recent call last):
File "main.py", line 110, in
main(opt)
File "main.py", line 78, in main
log_dict_train, _ = trainer.train(epoch, train_loader)
File "/content/drive/My Drive/CenterFace.pytorch/src/lib/trains/base_trainer.py", line 118, in train
return self.run_epoch('train', epoch, data_loader)
File "/content/drive/My Drive/CenterFace.pytorch/src/lib/trains/base_trainer.py", line 60, in run_epoch
for iter_id, batch in enumerate(data_loader):
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 363, in next
data = self._next_data()
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 989, in _next_data
return self._process_data(data)
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 1014, in _process_data
data.reraise()
File "/usr/local/lib/python3.6/dist-packages/torch/_utils.py", line 395, in reraise
raise self.exc_type(msg)
AttributeError: Caught AttributeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/_utils/worker.py", line 185, in _worker_loop
data = fetcher.fetch(index)
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/_utils/fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/content/drive/My Drive/CenterFace.pytorch/src/lib/datasets/sample/multi_pose.py", line 50, in getitem
img, anns = Data_anchor_sample(img, anns)
File "/content/drive/My Drive/CenterFace.pytorch/src/lib/utils/utils.py", line 38, in Data_anchor_sample
height, width, _ = image.shape
AttributeError: 'NoneType' object has no attribute 'shape'