diff --git a/datasets/floor/data.yaml b/datasets/floor/data.yaml new file mode 100644 index 0000000..ee55a9b --- /dev/null +++ b/datasets/floor/data.yaml @@ -0,0 +1,30 @@ +train: train/images +val: valid/images +test: test/images + +nc: 22 +names: + - floor + - corridor + - sidewalk + - chair + - table + - sofa_bed + - door + - elevator + - stairs + - wall + - window + - cabinet + - trash_can + - person + - bag + - electronics + - plant + - obstacle + - appliance + - toilet + - sink + - tableware + +path: /home/rongye/ProgramFiles/Yolo/datasets/floor \ No newline at end of file diff --git a/floor_finetune.md b/floor_finetune.md new file mode 100644 index 0000000..2738b28 --- /dev/null +++ b/floor_finetune.md @@ -0,0 +1,11 @@ +```bash +yolo segment train \ + model=yolo11l-seg-indoor.pt \ + data=datasets/floor/data.yaml \ + epochs=10 \ + freeze=20 \ + amp=False \ + device=1 \ + project=/home/rongye/ProgramFiles/Yolo/blind_guide_project \ + name=floor_finetune +``` \ No newline at end of file