• 카테고리

    질문 & 답변
  • 세부 분야

    컴퓨터 비전

  • 해결 여부

    미해결

yolov5 학습 에러

22.07.20 21:15 작성 조회수 1.12k

0

안녕하세요 강사님 수업 잘 듣고 있습니다. 다름아니라 제가 지금 yolov5로 이용하여 졸업작품을 만들고 있는 학생입니다. 그런데 데이터셋을 AIhub의 COCO 데이터로 가져왔고 그걸 YOLO 데이터셋으로 바꾸었습니다. yaml파일도 만들어서 했는데 

c:\Users\DBIR\AppData\Local\Programs\Python\Python39\lib\site-packages\setuptools\distutils_patch.py:25: UserWarning: Distutils was imported before Setuptools. This usage is discouraged and may exhibit undesirable behaviors or errors. Please use Setuptools' objects directly or at least import Setuptools first.

warnings.warn( train: weights=yolov5x.pt, cfg=D:\fish_detection\yolov5\models\yolov5x.yaml, data=D:\fish_detection\fish_Dataset\fish_detection.yaml, hyp=yolov5\data\hyps\hyp.scratch-low.yaml, epochs=100, batch_size=500, imgsz=2704, rect=False, resume=False, nosave=False, noval=False, noautoanchor=False, noplots=False, evolve=None, bucket=, cache=None, image_weights=False, device=, multi_scale=False, single_cls=False, optimizer=SGD, sync_bn=False, workers=8, project=yolov5\runs\train, name=yolov5_coco, exist_ok=False, quad=False, cos_lr=False, label_smoothing=0.0, patience=100, freeze=[0], save_period=-1, seed=0, local_rank=-1, entity=None, upload_dataset=False, bbox_interval=-1, artifact_alias=latest github: YOLOv5 is out of date by 12 commits. Use `git pull` or `git clone https://github.com/ultralytics/yolov5` to update. requirements: psutil not found and is required by YOLOv5, attempting auto-update... Requirement already satisfied: psutil in c:\users\dbir\anaconda3\pkgs\python-3.8.13-h6244533_0\lib\site-packages (5.9.1) requirements: 1 package updated per D:\fish_detection\yolov5\requirements.txt requirements: Restart runtime or rerun command for updates to take effect YOLOv5 v6.1-295-gf8722b4 Python-3.9.1 torch-1.10.0+cu102 CUDA:0 (GeForce RTX 2080 SUPER, 8192MiB) hyperparameters: lr0=0.01, lrf=0.01, momentum=0.937, weight_decay=0.0005, warmup_epochs=3.0, warmup_momentum=0.8, warmup_bias_lr=0.1, box=0.05, cls=0.5, cls_pw=1.0, obj=1.0, obj_pw=1.0, iou_t=0.2, anchor_t=4.0, fl_gamma=0.0, hsv_h=0.015, hsv_s=0.7, hsv_v=0.4, degrees=0.0, translate=0.1, scale=0.5, shear=0.0, perspective=0.0, flipud=0.0, fliplr=0.5, mosaic=1.0, mixup=0.0, copy_paste=0.0 Weights & Biases: run 'pip install wandb' to automatically track and visualize YOLOv5 runs (RECOMMENDED) TensorBoard: Start with 'tensorboard --logdir yolov5\runs\train', view at http://localhost:6006/ Traceback (most recent call last): File "D:\fish_detection\yolov5\train.py", line 667, in <module> main(opt) File "D:\fish_detection\yolov5\train.py", line 562, in main train(opt.hyp, opt, device, callbacks) File "D:\fish_detection\yolov5\train.py", line 106, in train data_dict = data_dict or check_dataset(data) # check if None File "D:\fish_detection\yolov5\utils\general.py", line 475, in check_dataset assert k in data, emojis(f"fish_detection.yaml '{k}:' field missing ❌") AssertionError: fish_detection.yaml 'train:' field missing

에러가 뜹니다. 에러 원인은 인터넷에서 찾아보고 있습니다.그런데 찾아도 답이 안 나와서 혹시 에러 원인을 알 수 있을까요?? 개발환경은 window입니다. 

답변 1

답변을 작성해보세요.

0

안녕하십니까, 

올려주신 메시지를 제대로 해석하기 힘들군요. 근데, 첫 메시지를 보면 error가 아니라 warning 입니다. 

나머지 부분도 error 메시지를 찾기가 힘든데, error 메시지만 캡쳐해 주실 수 있나요? 

그리고 windows 환경이 아니라 linux환경에서 함 테스트 해보실 수 있나요?

감사합니다.