RuntimeError: Net에 대한 state_dict를 로드하는 중 오류가 발생했습니다. 분류자의 크기가 일치하지 않습니다.

yolov4_deepsort 코드를 실행할 때 오류가 발생합니다:

RuntimeError: Net에 대한 state_dict 로드 중 오류:
    classifier.4.weight의 크기 불일치: 체크포인트에서 torch.Size([751, 256]) 모양의 매개변수 복사 중, 현재 모델의 모양은 torch.Size([ 10, 256]).
    classifier.4.bias의 크기 불일치: 체크포인트에서 torch.Size([751]) 모양이 있는 매개변수를 복사하는 중, 현재 모델의 모양은 torch.Size([10])입니다.

 오류가 발생하는 이유는 분류기 차원이 일치하지 않기 때문입니다. 10을 751로 변경하면 됩니다.

코드 위치 변경: deep_sort/deep/model.py 49행

class Net(nn.Module):
    def __init__(self, num_classes=751, reid=False):  
    #  将原来的num_classes=10改成num_classes=751即可

변경 후 정상적으로 실행됩니다.

https://github.com/ZQPei/deep_sort_pytorch/issues/134