Skip to content

Error when running the example_feat_extract.py #10

@berryweinst

Description

@berryweinst

Hi,
Maybe I'm missing something but after following the 3 steps + cloning the 'TF_FeatureExtraction' git itself. I run:
python example_feat_extract.py --network resnet_v1_50 --checkpoint ../resnet_v1_50.ckpt --image_path ../../studies/ECoG/video_data/frames/ --out_file ../features.h5 --num_classes 1000 --layer_names resnet_v1_50/logits

and getting this error:
Traceback (most recent call last):
File "example_feat_extract.py", line 24, in
from feature_extractor.feature_extractor import FeatureExtractor
File "/Users/berryweinstein/tensorflow/TF_FeatureExtraction/feature_extractor/feature_extractor.py", line 20, in
from nets import nets_factory
ModuleNotFoundError: No module named 'nets'

Can you please advise?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions