|
0
|
1 <tool id="yolo" name="YOLO" version="1.0">
|
|
|
2 <description>real-time object detection</description>
|
|
|
3 <requirements>
|
|
|
4 <requirement type="package" version="1.0">darknet</requirement>
|
|
|
5 </requirements>
|
|
|
6 <command detect_errors="exit_code"><![CDATA[
|
|
9
|
7 mkdir input_dir &&
|
|
|
8 mkdir output_dir &&
|
|
8
|
9 cp -R /home/greg/_conda/envs/__darknet@1.0/bin/* . &&
|
|
13
|
10 #for $i in $input:
|
|
14
|
11 #set input_filename = $i.file_name
|
|
15
|
12 #set full_name = $i.name
|
|
|
13 #set head = $full_name.split('.')[0]
|
|
|
14 #set output_filename = "%s_predictions.png" % $head
|
|
|
15 ln -s $input_filename input_dir/$full_name &&
|
|
|
16 darknet detect cfg/yolo.cfg yolo.weights 'input_dir/$full_name' -thresh $thresh 2> /dev/null &&
|
|
12
|
17 mv ./predictions.png output_dir/$output_filename
|
|
9
|
18 #end for
|
|
0
|
19 ]]></command>
|
|
|
20 <inputs>
|
|
8
|
21 <param name="input" format="jpg" type="data_collection" collection_type="list" label="Collection of image files" />
|
|
6
|
22 <param name="thresh" type="float" value="0.25" label="Object detection threshold" />
|
|
0
|
23 </inputs>
|
|
|
24 <outputs>
|
|
8
|
25 <collection name="output" type="list">
|
|
9
|
26 <discover_datasets pattern="__name__" directory="output_dir" format="png" />
|
|
8
|
27 </collection>
|
|
0
|
28 </outputs>
|
|
|
29 <tests>
|
|
|
30 <test>
|
|
|
31 </test>
|
|
|
32 </tests>
|
|
|
33 <help>
|
|
|
34 **What it does**
|
|
|
35
|
|
|
36 You only look once (YOLO) is a state-of-the-art, real-time object detection system.
|
|
|
37
|
|
|
38 -----
|
|
|
39
|
|
|
40 **Options**
|
|
|
41
|
|
|
42 </help>
|
|
|
43 <citations>
|
|
|
44 <citation type="bibtex">
|
|
|
45 @misc{darknet13,
|
|
|
46 author = {Joseph Redmon},
|
|
|
47 title = {Darknet: Open Source Neural Networks in C},
|
|
|
48 url = {http://pjreddie.com/darknet/},
|
|
|
49 year = {2013--2016}}
|
|
|
50 </citation>
|
|
|
51 <citation type="bibtex">
|
|
|
52 @article{redmon2016yolo9000,
|
|
|
53 title={YOLO9000: Better, Faster, Stronger},
|
|
|
54 author={Redmon, Joseph and Farhadi, Ali},
|
|
|
55 journal={arXiv preprint arXiv:1612.08242},
|
|
|
56 year={2016}}
|
|
|
57 </citation>
|
|
|
58 </citations>
|
|
|
59 </tool>
|
