基于YOLO格式标注文件,在原图片上画出标记框。可帮助检查标注的正确性,并将含有错标的原图像取出,您可以使用makesense.ai等工具重新进行标注。
- 将所有原始图片放入
./raw_images/
文件夹。 - 将所有标注文件(YOLO格式txt)放入
./labels/
文件夹。 - 将你的类别信息写入classes.txt,依次每个一行。
- 在terminal中运行
python draw_box.py
。
所有画出了标记框的图片现已被存至./save_image/
文件夹。
- 将含有错标的图片放入
./wrong/
文件夹. - 清空
./save_image/
文件夹. - 在terminal中运行
python get_origin_image.py
。
所有对应的原始图片现已被存至./save_image/
文件夹。
感谢批量将yolo-v3检测结果在原图上画矩形框显示提供的框架。在此基础上我改进了易用性,增加了对多类别标签的处理与标注,并增加了提取原图片的功能。