search

pjh5672 / YOLOv1

YOLOv1 implementation using PyTorch
GNU General Public License v3.0
6 stars 1 forks source link

Model Training #2

Closed joonesmks closed 1 year ago

joonesmks commented 1 year ago

Currently, the README.md file only outlines how to use the pre-trained weights to train a custom model.

It would be great, if the README would have a subsection that briefly outlines how to train the model on the PASCAL-VOC dataset in order to obtain training data like loss, etc.

pjh5672 commented 1 year ago

@joonesmks Thank for your interest 👍 I will add training description in more details in README.md soon. In other hand, you can train YOLOv1 model with [2-1. Model Training] subsection. For example of VOC training, you can set VOC data directory to "PATH" in "data/voc.yaml" in this repo. (PASCAL-VOC data directory must contain the sub-directory of "images/train2007", "images/train2012", "images/val2007", and "images/val2012".)

pjh5672 commented 1 year ago

@joonesmks for download proper VOC dataset, you can use this scripts with L79-84 from "https://github.com/ultralytics/yolov5/blob/master/data/VOC.yaml"