Hugging Face's logo

speedpointer
/
Kfoods

Model card Files
xet
 Community
Kfoods
File size: 2,194 Bytes
d658572
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
# no package for independent configuration management

# Initialize the parameters
CONF_THRES = 0.1 #0.5  # Confidence threshold
NMS_THRES  = 0.1 #0.4  # Non-maximum suppression threshold

INPWIDTH  = 416 # 32*10  # 608     #Width of network's input image # 320(32*10)
INPHEIGHT = 416 #32*9 # 608     #Height of network's input image # 288(32*9) best

# start video frame number
Video_Start_Frame = (21-4)*60*30+(2-39)*30 # compute the first starting location from a video

# model base dir
#ModelBaseDir = "C:/Users/mmc/workspace/AI_core/food_classification/yolo"
ModelBaseDir = "./yolo"
TEST_IMAGE_PATH ="/home/rtdatum/workspace/food_classification/images_rec/13118.jpg"
#TEST_IMAGE_PATH ="C:/Users/mmc/workspace/AI_core/food_classification/yolo/data/์‘ฅ๊ฐœ๋–ก/A240213_111121_0005.jpg"
#TEST_IMAGE_PATH ="./yolo/data/food/images/A270309_111112_0002.jpg"
TEST_VIDEO_PATH = \
    ""#"E:/Topes_data_related/์‹œ๋‚˜๋ฆฌ์˜ค ์˜์ƒ/์‹œ๋‚˜๋ฆฌ์˜ค ์˜์ƒ/20200909PM/6085-20200909-170439-1599638679.mp4"
SHOW_TEXT_FLAG = 1
PS_FLAG = 1

# Load names of classes, please don't include the first directory separator like "/data/..."
CLASSES_FILE = "data/food/food-classes.names"
CLASSES_FILE_CODE = "data/food/food-classes.codes"

# Give the configuration and weight files for the model and load the network using them.
# Don't include the first directory separator
# -- yolov3 ------
# ------- 3 layers
# itms
# Model_Configuration = "config/itms-dark-yolov3-tiny_3l-v3-2.cfg"
# Model_Weights = "data/food/weights/itms-dark-yolov3-tiny_3l-v3-2_200000.weights"
# food
Model_Configuration = "config/food-dark-yolov3-tiny_3l-v3-2.cfg"
Model_Weights = "data/food/weights/food-dark-yolov3-tiny_3l-v3-2_24000.weights"
# ------- full layers
#Model_Configuration = "config/food-dark-yolov3-full-2.cfg"
# Model_Weights = "data/food/weights/food-dark-yolov3-full-2_100000.weights"
# -- yolov4 -------
# 3l layers
# Model_Configuration = "config/food-dark-yolov4-tiny-3l-v1.cfg"
# Model_Weights       = "data/food/weights/food-dark-yolov4-tiny-3l-v1_best.weights"
# full layers
# Model_Configuration = "config/food-dark-yolov4-full.cfg"
# Model_Weights       = "data/food/weights/food-dark-yolov4-full_10000.weights"