pytorch/vision

Resnet34 as backbone for segmentation models

Open

#2 120 ouverte le 20 avr. 2020

Voir sur GitHub
 (5 commentaires) (0 réactions) (0 assignés)Python (15 050 stars) (6 858 forks)batch import
enhancementhelp wantedmodule: modelstopic: semantic segmentation

Description

🚀 Feature

I would like segmentation models with resnet34 backbone

Motivation

They have added resnet50 backbone.

Guide contributeur

Stack technique
python
Domaine
machine learning
Type d'issue
feature
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-2 days
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
stale
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
mostly clear
Prérequis
Python programmingPyTorch basicsUnderstanding of ResNet architecture
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
55
Direction de recherche
First, study how ResNet50 backbone was integrated into segmentation models in this repo (likely in torchvision/models/segmentation/backbone.py). Look for the existing FCN and DeepLab implementations to understand the pattern for adding a new backbone. Then, implement ResNet34 following similar steps: define the backbone, add it to the model registry, and update the weight loading logic if applicable. Check the issue comments for any additional guidance or unresolved questions.