SpursLipu / YOLOv3v4-ModelCompression-MultidatasetTraining-Multibackbone

YOLO ModelCompression MultidatasetTraining
GNU General Public License v3.0
445 stars 136 forks source link

在yolov3-mobilenet网络检测的部分,再添加一个yolo检测层,变为四个检测层,anchors也改为12组,但报错,不能训练。 #42

Closed chenjunson closed 4 years ago

chenjunson commented 4 years ago

我又尝试在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