Spaces:
Running
Running
| #------------------------------detect.py--------------------------------# | |
| # 这一部分是为了半自动标注数据，可以减轻负担，需要提前训练一个权重，以Labelme格式保存 | |
| # dir_origin_path 图片存放位置 | |
| # dir_save_path Annotation保存位置 | |
| # ----------------------------------------------------------------------# | |
| dir_detect_path: ./JPEGImages | |
| detect_save_path: ./Annotation | |
| # ----------------------------- train.py -------------------------------# | |
| nc: 8 # 类别的数量 | |
| classes: ["up","down","left","right","front","back","clockwise","anticlockwise"] # 类别 | |
| confidence: 0.5 # 置信度 | |
| nms_iou: 0.3 | |
| letterbox_image: False | |
| lr_decay_type: cos # 使用到的学习率下降方式，可选的有step、cos | |
| # 用于设置是否使用多线程读取数据 | |
| # 开启后会加快数据读取速度，但是会占用更多内存 | |
| # 内存较小的电脑可以设置为2或者0，win建议设为0 | |
| num_workers: 4 |