Skip to content

Instantly share code, notes, and snippets.

@SeanSyue

SeanSyue/yolo_gen_train_lst.py

Created August 14, 2018 02:56
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save SeanSyue/5b0a4d61254e27e021032378c3ee32ff to your computer and use it in GitHub Desktop.
Save SeanSyue/5b0a4d61254e27e021032378c3ee32ff to your computer and use it in GitHub Desktop.
Download ZIP
Generate training list for darknet yolo training.
Raw
yolo_gen_train_lst.py
from pathlib import Path
for img_file in Path('./JPEGImages').iterdir():
with open('train.txt', 'w') as f:
print(img_file.resolve(), file=f)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment