search

mkang315 / ASF-YOLO

[IMAVIS] Official implementation of "ASF-YOLO: A Novel YOLO Model with Attentional Scale Sequence Fusion for Cell Instance Segmentation".
GNU Affero General Public License v3.0
67 stars 3 forks source link

How to train own datasets, and how to get labels? #15

Open Xiaoliang-Liu666 opened 4 months ago

mkang315 commented 4 months ago
  1. Prepare your dataset with .txt format labels that follows the YOLO convention. You may use bounding box annotation tool software to get labels.
  2. Modify configure settings in relevant files, e.g., ./data/xxx.yaml, train.py.
  3. Install requirements.txt
  4. Training CLI python segment/train.py