423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-17 04:40:35 +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

2
projects/implicitron_trainer/configs/repro_base.yaml
View File

@@ -64,7 +64,7 @@ generic_model_args:
view_pooler_args:
view_sampler_args:
masked_sampling: false
image_feature_extractor_args:
image_feature_extractor_ResNetFeatureExtractor_args:
stages:
- 1
- 2

4
projects/implicitron_trainer/configs/repro_feat_extractor_normed.yaml
View File

@@ -1,6 +1,6 @@
generic_model_args:
image_feature_extractor_enabled: true
image_feature_extractor_args:
image_feature_extractor_class_type: ResNetFeatureExtractor
image_feature_extractor_ResNetFeatureExtractor_args:
add_images: true
add_masks: true
first_max_pool: true

4
projects/implicitron_trainer/configs/repro_feat_extractor_transformer.yaml
View File

@@ -1,6 +1,6 @@
generic_model_args:
image_feature_extractor_enabled: true
image_feature_extractor_args:
image_feature_extractor_class_type: ResNetFeatureExtractor
image_feature_extractor_ResNetFeatureExtractor_args:
add_images: true
add_masks: true
first_max_pool: false

4
projects/implicitron_trainer/configs/repro_feat_extractor_unnormed.yaml
View File

@@ -1,6 +1,6 @@
generic_model_args:
image_feature_extractor_enabled: true
image_feature_extractor_args:
image_feature_extractor_class_type: ResNetFeatureExtractor
image_feature_extractor_ResNetFeatureExtractor_args:
stages:
- 1
- 2