I cannot find a configuration for the large model for segmentation. Is it possible to

<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clip

V7x-seg about yolov7 HOT 1 OPEN

eafpres commented on June 18, 2024

V7x-seg

from yolov7.

Comments (1)

eafpres commented on June 18, 2024

python segment/train.py --epochs 2 --workers 4 --device 0 --batch-size 1 --data data/my_segmentation.yaml --img-size 1024 --hyp data/hyps/hyp.scratch-high-custom.yaml --cfg models/segment/yolov7x.yaml --weights yolov7x-seg.pt --optimizer 'AdamW' --name 20240312_seg_01

Traceback (most recent call last):
  File "/mnt/c/eaf_llc/aa-analytics_and_bi/autonet_torsten_ostervall/yolov7/seg/segment/train.py", line 681, in <module>
    main(opt)
  File "/mnt/c/eaf_llc/aa-analytics_and_bi/autonet_torsten_ostervall/yolov7/seg/segment/train.py", line 577, in main
    train(opt.hyp, opt, device, callbacks)
  File "/mnt/c/eaf_llc/aa-analytics_and_bi/autonet_torsten_ostervall/yolov7/seg/segment/train.py", line 266, in train
    compute_loss = ComputeLoss(model, overlap=overlap)  # init loss class
  File "/mnt/c/eaf_llc/aa-analytics_and_bi/autonet_torsten_ostervall/yolov7/seg/./utils/segment/loss.py", line 40, in __init__
    self.nm = m.nm  # number of masks
  File "/home/eafpres/.local/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1269, in __getattr__
    raise AttributeError("'{}' object has no attribute '{}'".format(
AttributeError: 'IDetect' object has no attribute 'nm'

I looked at the IDetect class in yolo.py, within the seg folder, after doing > git checkout u7 but it indeeds has no attribute nm

from yolov7.

Recommend Projects