Skip to content

After Infer.py : the result is the only black picture. What is the problem? #95

@Ratatou2

Description

@Ratatou2

I try this cods

CUDA_VISIBLE_DEVICES=1 python train.py --experiment_dir=experiment
--experiment_id=0
--batch_size=32
--lr=0.001
--epoch=30
--sample_steps=100
--schedule=10
--L1_penalty=100
--Lconst_penalty=15

=> I can confirm that my model was trained correctly through the sample picture

CUDA_VISIBLE_DEVICES=1 python infer.py --model_dir=/home/zi2zi/experiment/checkpoint/experiment_0_batch_32
--batch_size=32
--source_obj=/home/zi2zi/experiment/data/val.obj
--embedding_ids=1
--save_dir=save_dir/

=> After executing infer.py, the 'save_dir' was filled with only black pictures....

do u know why??

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