423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-14 11:26:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make feature extractor pluggable

Browse Source 
Summary: Make ResNetFeatureExtractor be an implementation of FeatureExtractorBase.

Reviewed By: davnov134

Differential Revision: D35433098

fbshipit-source-id: 0664a9166a88e150231cfe2eceba017ae55aed3a
This commit is contained in:
Jeremy Reizenstein
2022-05-18 08:50:18 -07:00
committed by Facebook GitHub Bot
parent cd7b885169
commit 9ec9d057cc
12 changed files with 151 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/implicitron/data/overrides.yaml
View File

@@ -17,7 +17,7 @@ sampling_mode_training: mask_sample
sampling_mode_evaluation: full_grid
raysampler_class_type: AdaptiveRaySampler
renderer_class_type: LSTMRenderer
image_feature_extractor_enabled: true
image_feature_extractor_class_type: ResNetFeatureExtractor
view_pooler_enabled: true
implicit_function_class_type: IdrFeatureField
loss_weights:
@@ -73,7 +73,7 @@ renderer_LSTMRenderer_args:
hidden_size: 16
n_feature_channels: 256
verbose: false
image_feature_extractor_args:
image_feature_extractor_ResNetFeatureExtractor_args:
name: resnet34
pretrained: true
stages: