Official LSM-YOLO

Implementation of paper - LSM-YOLO: A Compact and Effective ROI Detector for Medical Detection

Installation

# pip install required packages
pip install -r requirements.txt

Training

# train after modifying the parameters in train.py file
python train.py 

Validating and Testing

# test after modifying the parameters in val.py file
python val.py

Inference

# inference after modifying the parameters in detect.py file
python detect.py