We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Hi
I currently have a custom dataset which i have organized like this
MAIN_FOLDER/yolo/data/tmp/images->train(contains images for training ) MAIN_FOLDER/yolo/data/tmp/images->val(contains images for validation)
MAIN_FOLDER/yolo/data/tmp/labels->train(contains labels for training ) MAIN_FOLDER/yolo/data/tmp/labels->val(contains labels for training )
And below is the contents of my custom file general.yaml which i have placed in: MAIN_FOLDER/yolo/config/dataset
Below is how my general.yaml looks like
==================================================================== path: MAIN_FOLDER/yolo/data/tmp train: train validation: val class_num: 2 class_list: ['A', 'B'] image_size: [640, 640] learning_rate: 0.0000001 out_path: runs exist_ok: True lucky_number: 10 use_wandb: True use_tensorboard: False
The command i use for training is python lazy.py task=train model=v9-m dataset=general device=gpu weight=False
below is my training config file train.yaml
============================================================================= task: train
defaults:
epoch: 10
data: batch_size: 5 image_size: ${image_size} cpu_num: ${cpu_num} shuffle: True pin_memory: True data_augment: Mosaic: 1 # MixUp: 1 # HorizontalFlip: 0.5 RandomCrop: 1 RemoveOutliers: 1e-8
optimizer: type: SGD args: lr: 0.00001 weight_decay: 0.0005 momentum: 0.937 nesterov: true
loss: objective: BCELoss: 0.5 BoxLoss: 7.5 DFLoss: 1.5 aux: 0.25 matcher: iou: CIoU topk: 10 factor: iou: 6.0 cls: 0.5
scheduler: type: LinearLR warmup: epochs: 3.0 args: total_iters: ${task.epoch} start_factor: 1 end_factor: 0.01
Any idea what i am doing wrong maybe it isn't even loading my data.
The text was updated successfully, but these errors were encountered:
No branches or pull requests
Hi
I currently have a custom dataset which i have organized like this
MAIN_FOLDER/yolo/data/tmp/images->train(contains images for training )
MAIN_FOLDER/yolo/data/tmp/images->val(contains images for validation)
MAIN_FOLDER/yolo/data/tmp/labels->train(contains labels for training )
MAIN_FOLDER/yolo/data/tmp/labels->val(contains labels for training )
And below is the contents of my custom file general.yaml which i have placed in:
MAIN_FOLDER/yolo/config/dataset
Below is how my general.yaml looks like
====================================================================
path: MAIN_FOLDER/yolo/data/tmp
train: train
validation: val
class_num: 2
class_list: ['A', 'B']
image_size: [640, 640]
learning_rate: 0.0000001
out_path: runs
exist_ok: True
lucky_number: 10
use_wandb: True
use_tensorboard: False
weight: True # Path to weight or True for auto, False for no pretrained weight
The command i use for training is python lazy.py task=train model=v9-m dataset=general device=gpu weight=False
below is my training config file train.yaml
=============================================================================
task: train
defaults:
epoch: 10
data:
batch_size: 5
image_size: ${image_size}
cpu_num: ${cpu_num}
shuffle: True
pin_memory: True
data_augment:
Mosaic: 1
# MixUp: 1
# HorizontalFlip: 0.5
RandomCrop: 1
RemoveOutliers: 1e-8
optimizer:
type: SGD
args:
lr: 0.00001
weight_decay: 0.0005
momentum: 0.937
nesterov: true
loss:
objective:
BCELoss: 0.5
BoxLoss: 7.5
DFLoss: 1.5
aux:
0.25
matcher:
iou: CIoU
topk: 10
factor:
iou: 6.0
cls: 0.5
scheduler:
type: LinearLR
warmup:
epochs: 3.0
args:
total_iters: ${task.epoch}
start_factor: 1
end_factor: 0.01
ema:
enable: true
decay: 0.995
Any idea what i am doing wrong maybe it isn't even loading my data.
The text was updated successfully, but these errors were encountered: