Откуда я могу получить подробное описание всех методов для модели в pytorch torchvison？ - PullRequest
Новая
       10

Откуда я могу получить подробное описание всех методов для модели в pytorch torchvison？

0 голосов
/ 01 августа 2020

Я новичок в pytorch ， когда я читаю исходный код проекта о маске rcnn. Я не знаю, откуда я могу получить некоторую информацию о некоторых методах, которые я не понимаю. В официальной документации нет t кажется очень подробным? 

# load an instance segmentation model pre-trained pre-trained on COCO
model = torchvision.models.detection.maskrcnn_resnet50_fpn(pretrained=True)
# get number of input features for the classifier
in_features = model.roi_heads.box_predictor.cls_score.in_features
# replace the pre-trained head with a new one
model.roi_heads.box_predictor = FastRCNNPredictor(in_features, num_classes)

Как и в коде выше, я не смог получить подробную информацию об атрибуте «roi_head» из do c модели. Откуда я могу узнать об этом？

1 Ответ

0 голосов
/ 01 августа 2020

Вы не найдете такого в документации. Вам придется погрузиться в исходный код. API обнаружения объектов, особенно двухэтапные подходы на основе привязки, немного сложны и, как правило, содержат слишком много компонентов и гиперпараметров. Команда PyTorch уже проделала невероятную работу, сделав этот API модульным и отчасти простым в использовании. В случае c roi_heads вы можете посмотреть здесь , чтобы узнать больше об этом. В общем, все компоненты можно найти в torchvision/models/detection.

В любом случае, вы всегда можете открыть вопрос и попросить их расширить документацию. Или мы можем сделать это сами и сделать pull request :)

...