DDGRCF / YOLOX_OBB

https://zhuanlan.zhihu.com/p/430850089
Apache License 2.0
147 stars 19 forks source link

搭建环境过程中的错误 #16

Open mortal-Zero opened 2 years ago

mortal-Zero commented 2 years ago

您好,我在使用您的代码搭建环境的时候,遇到了问题,我的电脑环境是这样的


看起来好像是缺少文件,希望您能帮助我。
DDGRCF commented 2 years ago

您好,非常抱歉。onnx部分我还没有完成,但是去掉onnx部分编译对整体不影响,你可以在setup中将其杠掉

mortal-Zero commented 2 years ago

您好,非常抱歉。onnx部分我还没有完成,但是去掉onnx部分编译对整体不影响,你可以在setup中将其杠掉

好的,了解了,感谢答复。

mortal-Zero commented 2 years ago

您好,非常抱歉。onnx部分我还没有完成,但是去掉onnx部分编译对整体不影响,你可以在setup中将其杠掉

不好意思,再打扰一下,您的这套代码支持COCO格式数据集训练吗?

DDGRCF commented 2 years ago

您好,非常抱歉。onnx部分我还没有完成,但是去掉onnx部分编译对整体不影响,你可以在setup中将其杠掉

不好意思,再打扰一下,您的这套代码支持COCO格式数据集训练吗?

你把coco格式转成dota格式,然后再使用BboxToolkit将格式转换为可以训练的格式就行了

mortal-Zero commented 2 years ago

您好,非常抱歉。onnx部分我还没有完成,但是去掉onnx部分编译对整体不影响,你可以在setup中将其杠掉

不好意思,再打扰一下,您的这套代码支持COCO格式数据集训练吗?

你把coco格式转成dota格式,然后再使用BboxToolkit将格式转换为可以训练的格式就行了

是的,我按照您所说的步骤完成了数据集的转换,但我开始训练后,它卡在了初始化阶段,辛苦问您一下可能是什么原因? 终端信息如下:

(yolox_obb) etop@etop-dl:/home/Data/Projects/YOLOX_OBB$ python tools/train.py -expn yoloxs_seal_20220411 -f exps/example/yolox_obb/yolox_obb_dota_seal.py -d 1 -b 8
2022-04-11 19:29:43 | INFO     | yolox.core.trainer:132 - args: Namespace(batch_size=8, cache=False, ckpt=None, devices=1, dist_backend='nccl', dist_url=None, exp_file='exps/example/yolox_obb/yolox_obb_dota_seal.py', experiment_name='yoloxs_seal_20220411', fp16=False, machine_rank=0, name=None, num_machines=1, occupy=False, options=None, resume=False, start_epoch=None)
2022-04-11 19:29:43 | INFO     | yolox.core.trainer:133 - exp value:
╒═══════════════════╤══════════════════════════════════════════════════════╕
│ keys              │ values                                               │
╞═══════════════════╪══════════════════════════════════════════════════════╡
│ seed              │ None                                                 │
├───────────────────┼──────────────────────────────────────────────────────┤
│ output_dir        │ 'YOLOX_outputs'                                      │
├───────────────────┼──────────────────────────────────────────────────────┤
│ print_interval    │ 10                                                   │
├───────────────────┼──────────────────────────────────────────────────────┤
│ eval_interval     │ 10                                                   │
├───────────────────┼──────────────────────────────────────────────────────┤
│ modules_config    │ 'configs/modules/yoloxs_obb.yaml'                    │
├───────────────────┼──────────────────────────────────────────────────────┤
│ losses_config     │ 'configs/losses/yolox_losses_obb.yaml'               │
├───────────────────┼──────────────────────────────────────────────────────┤
│ dataset_config    │ 'configs/datasets/dota_seal.yaml'                    │
├───────────────────┼──────────────────────────────────────────────────────┤
│ data_num_workers  │ 12                                                   │
├───────────────────┼──────────────────────────────────────────────────────┤
│ input_size        │ (640, 640)                                           │
├───────────────────┼──────────────────────────────────────────────────────┤
│ multiscale_range  │ 5                                                    │
├───────────────────┼──────────────────────────────────────────────────────┤
│ mosaic_prob       │ 1.0                                                  │
├───────────────────┼──────────────────────────────────────────────────────┤
│ mixup_prob        │ 1.0                                                  │
├───────────────────┼──────────────────────────────────────────────────────┤
│ hsv_prob          │ 1.0                                                  │
├───────────────────┼──────────────────────────────────────────────────────┤
│ flip_prob         │ 0.5                                                  │
├───────────────────┼──────────────────────────────────────────────────────┤
│ degrees           │ 10.0                                                 │
├───────────────────┼──────────────────────────────────────────────────────┤
│ translate         │ 0.1                                                  │
├───────────────────┼──────────────────────────────────────────────────────┤
│ mosaic_scale      │ (0.4, 1.2)                                           │
├───────────────────┼──────────────────────────────────────────────────────┤
│ mixup_scale       │ (0.4, 1.2)                                           │
├───────────────────┼──────────────────────────────────────────────────────┤
│ shear             │ 2                                                    │
├───────────────────┼──────────────────────────────────────────────────────┤
│ perspective       │ 0.0                                                  │
├───────────────────┼──────────────────────────────────────────────────────┤
│ enable_mixup      │ True                                                 │
├───────────────────┼──────────────────────────────────────────────────────┤
│ warmup_epochs     │ 1                                                    │
├───────────────────┼──────────────────────────────────────────────────────┤
│ max_epoch         │ 100                                                  │
├───────────────────┼──────────────────────────────────────────────────────┤
│ warmup_lr         │ 0                                                    │
├───────────────────┼──────────────────────────────────────────────────────┤
│ basic_lr_per_img  │ 0.00015625                                           │
├───────────────────┼──────────────────────────────────────────────────────┤
│ scheduler         │ 'yoloxwarmcos'                                       │
├───────────────────┼──────────────────────────────────────────────────────┤
│ no_aug_epochs     │ 15                                                   │
├───────────────────┼──────────────────────────────────────────────────────┤
│ min_lr_ratio      │ 0.05                                                 │
├───────────────────┼──────────────────────────────────────────────────────┤
│ ema               │ True                                                 │
├───────────────────┼──────────────────────────────────────────────────────┤
│ no_eval           │ False                                                │
├───────────────────┼──────────────────────────────────────────────────────┤
│ weight_decay      │ 0.0005                                               │
├───────────────────┼──────────────────────────────────────────────────────┤
│ momentum          │ 0.9                                                  │
├───────────────────┼──────────────────────────────────────────────────────┤
│ exp_name          │ 'yoloxs_seal_20220411'                               │
├───────────────────┼──────────────────────────────────────────────────────┤
│ test_size         │ (640, 640)                                           │
├───────────────────┼──────────────────────────────────────────────────────┤
│ postprocess_cfg   │ {'conf_thre': 0.01, 'nms_thre': 0.65}                │
├───────────────────┼──────────────────────────────────────────────────────┤
│ data_dir          │ 'datasets/DOTA1_0'                                   │
├───────────────────┼──────────────────────────────────────────────────────┤
│ train_ann         │ 'trainval'                                           │
├───────────────────┼──────────────────────────────────────────────────────┤
│ val_ann           │ 'val'                                                │
├───────────────────┼──────────────────────────────────────────────────────┤
│ num_classes       │ 4                                                    │
├───────────────────┼──────────────────────────────────────────────────────┤
│ copy_paste_prob   │ 1.0                                                  │
├───────────────────┼──────────────────────────────────────────────────────┤
│ enable_copy_paste │ True                                                 │
├───────────────────┼──────────────────────────────────────────────────────┤
│ enable_debug      │ False                                                │
├───────────────────┼──────────────────────────────────────────────────────┤
│ enable_resample   │ True                                                 │
├───────────────────┼──────────────────────────────────────────────────────┤
│ aug_ignore        │ None                                                 │
├───────────────────┼──────────────────────────────────────────────────────┤
│ empty_ignore      │ True                                                 │
├───────────────────┼──────────────────────────────────────────────────────┤
│ evaluate_cfg      │ {'is_merge': False, 'is_submiss': False, 'nproc': 1} │
├───────────────────┼──────────────────────────────────────────────────────┤
│ test_ann          │ 'test'                                               │
╘═══════════════════╧══════════════════════════════════════════════════════╛
2022-04-11 19:29:43 | INFO     | yolox.models.parse_model:17 - overriding modules.yaml num_classes=80 with num_classes=4
/home/etop/miniconda3/envs/yolox_obb/lib/python3.7/site-packages/torch/functional.py:445: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at  ../aten/src/ATen/native/TensorShape.cpp:2157.)
  return _VF.meshgrid(tensors, **kwargs)  # type: ignore[attr-defined]
2022-04-11 19:29:43 | INFO     | yolox.core.trainer:139 - Model Summary: Params: 8.05M, Gflops: 21.68
2022-04-11 19:29:47 | INFO     | yolox.core.trainer:157 - init prefetcher, this might take one minute or less...
DDGRCF commented 2 years ago

请问卡了多久,是一直卡吗。。。

mortal-Zero commented 2 years ago

请问卡了多久,是一直卡吗。。。

是的,它停留在这了一整晚。