MG31 commited on
Commit
0122bc9
1 Parent(s): a451ba6

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +23 -0
config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "task": "object-detection",
3
+ "pipeline_tag": "object-detection",
4
+ "model_type": "yolos",
5
+ "framework": "pytorch",
6
+ "model_format": "pt",
7
+ "input_format": "image",
8
+ "output_format": "object-detection",
9
+ "input_size": [640, 640],
10
+ "labels": [
11
+ "Road_No_Parking",
12
+ "Road_Speed_Limit_in_School_Zone",
13
+ "Road_School_Zone",
14
+ "Crosswalk",
15
+ "Road_No_Stopping_or_Parking",
16
+ "Road_No_Stopping_Zone",
17
+ "stop",
18
+ "traffic_lane_yellow_solid",
19
+ "school_zone",
20
+ "no_parking",
21
+ "fire_hydrant"
22
+ ]
23
+ }