Mercurial > repos > greg > yolo
changeset 9:0206a6a1b89e draft
Uploaded
| author | greg |
|---|---|
| date | Tue, 17 Oct 2017 10:55:45 -0400 |
| parents | da04a9890937 |
| children | b087daa22607 |
| files | yolo.xml |
| diffstat | 1 files changed, 11 insertions(+), 4 deletions(-) [+] |
line wrap: on
line diff
--- a/yolo.xml Tue Oct 17 10:40:23 2017 -0400 +++ b/yolo.xml Tue Oct 17 10:55:45 2017 -0400 @@ -4,10 +4,17 @@ <requirement type="package" version="1.0">darknet</requirement> </requirements> <command detect_errors="exit_code"><![CDATA[ -mkdir output && +mkdir input_dir && +mkdir output_dir && cp -R /home/greg/_conda/envs/__darknet@1.0/bin/* . && -darknet detect cfg/yolo.cfg yolo.weights '$input' -thresh $thresh 2> /dev/null -&& mv ./predictions.png output +#for $index, $i in enumerate('$input'): + #set input_filename = $i.file_name + #set name = $i.name + #set output_filename = %05d_predictions.png % index + ln -s $input_filename input_dir/$name && + darknet detect cfg/yolo.cfg yolo.weights 'input_dir/$name' -thresh $thresh 2> /dev/null && + mv ./predictions.png output_dir/output_filename +#end for ]]></command> <inputs> <param name="input" format="jpg" type="data_collection" collection_type="list" label="Collection of image files" /> @@ -15,7 +22,7 @@ </inputs> <outputs> <collection name="output" type="list"> - <discover_datasets pattern="__name__" directory="output" format="png" /> + <discover_datasets pattern="__name__" directory="output_dir" format="png" /> </collection> </outputs> <tests>