Major Updates
- refactorize main predict function:
bash
python scripts/predict.py --visual --source image/file/or/folder --model_path path/to/model --config_path path/to/config
will export predictions visuals to `runs/predict/exp` folder.
If you want to export prediction pickles and cropped predictions add `--pickle` and `--crop` arguments.
- add option to export coco json:
bash
python scripts/predict.py --coco_file path/to/coco/file --source coco/images/directory --model_path path/to/model --config_path path/to/config
will perform inference using provided coco file, then export results as a coco json file to `runs/predict/exp/results.json`
- add script for coco2yolov5 conversion:
bash
python scripts/coco2yolov5.py --coco_file path/to/coco/file --source coco/images/directory --train_split 0.9
will convert given coco dataset to yolov5 format and export to `runs/coco2yolov5/exp` folder .
Minor Updates
- add export_format to crop_object_predictions
- fix coco annotation serialization
- more generic coco handling
- add option to export visuals
- update ci/package testing workflows