Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/
Traceback (most recent call last):
File "train.py", line 512, in
train(hyp) # train normally
File "train.py", line 94, in train
model = Darknet(cfg, quantized=opt.quantized, qlayers=opt.qlayers).to(device)
File "/home/chenjunsong/U-YOLOv3/models.py", line 364, in init
qlayers=self.qlayers)
File "/home/chenjunsong/U-YOLOv3/models.py", line 233, in create_modules
stride=stride[yolo_index])
IndexError: list index out of range
我又尝试在yolov3后面再添加了个yolo检测层,结果还是同样的报错, 报错如下:chenjunsong@chenjunsong-GJ5CN64:~/U-YOLOv3$ python3 train.py --data data/obj.data --batch-size 6 --cfg cfg/yolov3-mobilenet/yolov3-mobilenet1.cfg Apex recommended for faster mixed precision training: https://github.com/NVIDIA/apex Namespace(KDstr=-1, accumulate=2, adam=False, batch_size=6, bucket='', cache_images=False, cfg='./cfg/yolov3-mobilenet/yolov3-mobilenet1.cfg', data='data/obj.data', device='', epochs=601, evolve=False, img_size=[320, 640], multi_scale=False, name='', nosave=False, notest=False, prune=-1, pt=False, qlayers=-1, quantized=-1, rect=False, resume=False, s=0.001, single_cls=False, sr=False, t_cfg='', t_weights='', weights='') Using CUDA device0 _CudaDeviceProperties(name='GeForce GTX 1060', total_memory=6072MB)
Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/ Traceback (most recent call last): File "train.py", line 512, in
train(hyp) # train normally
File "train.py", line 94, in train
model = Darknet(cfg, quantized=opt.quantized, qlayers=opt.qlayers).to(device)
File "/home/chenjunsong/U-YOLOv3/models.py", line 364, in init
qlayers=self.qlayers)
File "/home/chenjunsong/U-YOLOv3/models.py", line 233, in create_modules
stride=stride[yolo_index])
IndexError: list index out of range