AttributeError: 'Detector' object has no attribute 'input_size'

[nemtiax]

I'm getting this error when training on COCO gets to the first evaluation step:

Traceback (most recent call last):
  File "train.py", line 229, in <module>
    dts = model_eval.detect_imgSeq(val_img_dir, input_size=target_size)
  File "/home/ubuntu/RAPiD_clean/RAPiD/api.py", line 90, in detect_imgSeq
    dts = self._detect_iter(iter(ims), **kwargs)
  File "/home/ubuntu/RAPiD_clean/RAPiD/api.py", line 97, in _detect_iter
    detections = self._predict_pil(pil_img=pil_frame, **kwargs)
  File "/home/ubuntu/RAPiD_clean/RAPiD/api.py", line 115, in _predict_pil
    input_size = kwargs.get('input_size', self.input_size)
AttributeError: 'Detector' object has no attribute 'input_size'

I think the reason this happens is that when a Detector is initialized for evaluation, it gets passed a model:

RAPiD/train.py

Line 226 in c6fabcf

model_eval = api.Detector(conf_thres=0.005, model=model)

If we look at Detector's init method, it sets self.input_size and self.conf_thres on lines 43-44, but if you pass a model, it will return on line 27, so input_size and conf_thres are never initialized.

Moving the initialization of input_size and conf_thres to the top of the init method resolved the issue for me, but I'm not sure if that's really the root cause, or if I'm just patching over some other problem (like maybe these are intended to be set elsewhere, and that's not working).

[duanzhiihao]

I think this problem shouldn't happen given that you specify the input_size=target_size when calling detect_imgSeq. Could you check if your target_size is None?

RAPiD/train.py

Line 227 in c6fabcf

dts = model_eval.detect_imgSeq(val_img_dir, input_size=target_size)

Moving the initialization of input_size and conf_thres to the top of the init method resolved the issue for me, but I'm not sure if that's really the root cause, or if I'm just patching over some other problem (like maybe these are intended to be set elsewhere, and that's not working).

This should work fine and cause no other problems.

[nemtiax]

I think my target_size is being set here:

RAPiD/train.py

Line 41 in c6fabcf

target_size = 1024 if args.high_resolution else 608

Which should propagate down to the call to detect_imgSeq. I'll add some debugging outputs and verify whether it's actually getting there, though.

[duanzhiihao]

It turns out that there is a bug in the api.py. I fixed it and it should work now. Could you try again and tell me if your problem resolved?