Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

python train.py

See original GitHub issue

when I use python train.py CenterNet-104 to train the model, there is such error Traceback (most recent call last): File "train.py", line 43, in prefetch_data data, ind = sample_data(db, ind, data_aug=data_aug) File "/media/disk/wudi/CenterNet/sample/coco.py", line 200, in sample_data return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug) File "/media/disk/wudi/CenterNet/sample/coco.py", line 100, in kp_detection image, detections = random_crop(image, detections, rand_scales, input_size, border=border) File "/media/disk/wudi/CenterNet/sample/utils.py", line 57, in random_crop image_height, image_width = image.shape[0:2] AttributeError: 'NoneType' object has no attribute 'shape' whe does the error happen? how to avoid this error? Thanks

Issue Analytics

State:
Created 4 years ago
Reactions:1
Comments:18 (1 by maintainers)

Top GitHub Comments

1reaction

nuist-xinyucommented, May 31, 2019

I just copied the image in the verification set directly to the training set.

1reaction

wudi00commented, May 22, 2019

@Duankaiwen Hi, I want to train my dataset, and where shouled I modify in the code?

Top Results From Across the Web

yolov5/train.py at master - GitHub

Train a YOLOv5 model on a custom dataset. Models and datasets download automatically from the latest YOLOv5 release. Usage - Single- ...

Tutorial: Train a first Python machine learning model (SDK v1)

Create training scripts. First you define the neural network architecture in a model.py file. All your training code will go into the src ......

yolov5 does not train the model - Stack Overflow

I trained a dataset by yolov5. The model is existed but it does not work. It does not predict excatly. Here the code...

train.py · RichardoMu/yolov5-mask-detection-python - Gitee.com

python path/to/train.py --data coco128.yaml --weights yolov5s.pt --img 640. """ import argparse. import logging. import math. import os. import random.

How to Train YOLO v5 on a Custom Dataset - Paperspace Blog

This tutorial will show you how to implement and train YOLOv5 on your own ... Hyperparameters for COCO training from scratch # python...