init

f350b742 · Quan · 33d07985 · f350b742 · 7defbe39 · f350b742
Commit f350b742 authored Dec 01, 2020 by Quan
218 changed files
--- a/README.md
+++ b/README.md
+# README
+* This project  implement the head detection model and human detection. More detail about each model can be read at [head detection](http://192.168.0.232:8929/tienln4/ai_camera_detector/-/blob/master/docs/head.md) and [person detection](http://192.168.0.232:8929/tienln4/ai_camera_detector/-/blob/master/docs/person.md)
+
+# Version
+- mb2-ssd-lite_f19: original ssd-lite model
+- mb2-ssd-lite_f38: for head detection
+- mb2-ssd-lite_f38_person: for person detection (small objects)
+- rfb_tiny_mb2_ssd: for person detection (there are two sub version: c32 fast and c64 slow)
+```
+* config(c32 and c64): at line 76 (/media/ducanh/DATA/tienln/ai_camera/ai_camera_detector/module/rfb_tiny_mobilenet_v2.py)   
+```
\ No newline at end of file
--- a/ai_camera_detector @ 7defbe39
+++ b/ai_camera_detector @ 7defbe39
-Subproject commit 7defbe39bdff3b81a02cec330119d2cef4459cf1
--- a/app/head/mb_ssd_lite_f38_150_193_14.dlc
+++ b/app/head/mb_ssd_lite_f38_150_193_14.dlc
--- a/app/head/mb_ssd_lite_f38_150_193_14.pth
+++ b/app/head/mb_ssd_lite_f38_150_193_14.pth
--- a/app/head/mb_ssd_lite_f38_150_193_14_quantize.dlc
+++ b/app/head/mb_ssd_lite_f38_150_193_14_quantize.dlc
--- a/app/person/rfb_tiny_mb2_ssd_c32/rfb_tiny_mb2_ssd_c32_63_208_222.dlc
+++ b/app/person/rfb_tiny_mb2_ssd_c32/rfb_tiny_mb2_ssd_c32_63_208_222.dlc
--- a/app/person/rfb_tiny_mb2_ssd_c32/rfb_tiny_mb2_ssd_c32_63_208_222.pth
+++ b/app/person/rfb_tiny_mb2_ssd_c32/rfb_tiny_mb2_ssd_c32_63_208_222.pth
--- a/app/person/rfb_tiny_mb2_ssd_c32/rfb_tiny_mb2_ssd_c32_63_208_222_quantize.dlc
+++ b/app/person/rfb_tiny_mb2_ssd_c32/rfb_tiny_mb2_ssd_c32_63_208_222_quantize.dlc
--- a/app/person/rfb_tiny_mb2_ssd_c64/rfb_tiny_mb2_ssd_c64_146_142_19.dlc
+++ b/app/person/rfb_tiny_mb2_ssd_c64/rfb_tiny_mb2_ssd_c64_146_142_19.dlc
--- a/app/person/rfb_tiny_mb2_ssd_c64/rfb_tiny_mb2_ssd_c64_146_142_19.onnx
+++ b/app/person/rfb_tiny_mb2_ssd_c64/rfb_tiny_mb2_ssd_c64_146_142_19.onnx
--- a/app/person/rfb_tiny_mb2_ssd_c64/rfb_tiny_mb2_ssd_c64_146_142_19.pth
+++ b/app/person/rfb_tiny_mb2_ssd_c64/rfb_tiny_mb2_ssd_c64_146_142_19.pth
--- a/app/person/rfb_tiny_mb2_ssd_c64/rfb_tiny_mb2_ssd_c64_146_142_19_quantize.dlc
+++ b/app/person/rfb_tiny_mb2_ssd_c64/rfb_tiny_mb2_ssd_c64_146_142_19_quantize.dlc
--- a/app/quantize/create_raw_data.py
+++ b/app/quantize/create_raw_data.py
+import sys
+sys.path.append('/media/ducanh/DATA/tienln/ai_camera/detector/')
+import cv2
+# from PIL import Image
+import numpy as np
+import os
+from datasets.data_preprocessing import PredictionTransform
+
+size = 300
+transform = PredictionTransform(size)
+
+def get_input_FD(img_raw):
+    img = transform(img_raw)
+    img = np.expand_dims(img, axis=0)
+    return img
+
+image_path = '/media/ducanh/DATA/tienln/ai_camera/app/app_head_detection/imgs'
+images = os.listdir(image_path)
+paths = []
+for image_name in images:
+    print(image_name)
+    img = cv2.imread(os.path.join(image_path, image_name))
+    img_processed = get_input_FD(img)
+    img_processed.tofile(os.path.join('/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data', image_name).replace('.jpg', ".raw"))
+    paths.append(os.path.join('/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data', image_name).replace('.jpg', ".raw"))
+    print("test")
+
+with open('/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/data_quantize_person.txt', 'w') as f:
+    for item in paths:
+        f.write("%s\n" % item)
\ No newline at end of file
--- a/app/quantize/data_list.txt
+++ b/app/quantize/data_list.txt
+raw_data/66063.raw
+raw_data/10118.raw
+raw_data/52309.raw
+raw_data/17374.raw
+raw_data/55038.raw
+raw_data/21594.raw
+raw_data/46879.raw
+raw_data/72073.raw
+raw_data/34577.raw
+raw_data/22842.raw
+raw_data/19030.raw
+raw_data/22111.raw
+raw_data/72055.raw
+raw_data/50707.raw
+raw_data/29345.raw
+raw_data/14942.raw
+raw_data/49723.raw
+raw_data/61017.raw
+raw_data/51772.raw
+raw_data/42987.raw
+raw_data/51994.raw
+raw_data/86928.raw
+raw_data/59368.raw
+raw_data/42194.raw
+raw_data/47692.raw
+raw_data/12384.raw
+raw_data/25929.raw
+raw_data/61103.raw
+raw_data/53785.raw
+raw_data/23324.raw
+raw_data/72880.raw
+raw_data/63513.raw
+raw_data/68275.raw
+raw_data/75792.raw
+raw_data/85954.raw
+raw_data/46905.raw
+raw_data/69667.raw
+raw_data/47457.raw
+raw_data/69533.raw
+raw_data/15378.raw
+raw_data/37103.raw
+raw_data/63752.raw
+raw_data/51264.raw
+raw_data/44684.raw
+raw_data/70829.raw
+raw_data/25730.raw
+raw_data/58802.raw
+raw_data/37585.raw
+raw_data/12155.raw
+raw_data/39834.raw
+raw_data/79228.raw
+raw_data/27457.raw
+raw_data/59429.raw
+raw_data/32689.raw
+raw_data/82567.raw
+raw_data/78851.raw
+raw_data/70275.raw
+raw_data/37352.raw
+raw_data/58027.raw
+raw_data/53293.raw
+raw_data/77275.raw
+raw_data/68371.raw
+raw_data/56403.raw
+raw_data/84186.raw
+raw_data/25011.raw
+raw_data/29547.raw
+raw_data/43464.raw
+raw_data/78356.raw
+raw_data/18074.raw
+raw_data/25941.raw
+raw_data/26285.raw
+raw_data/52162.raw
+raw_data/44600.raw
+raw_data/21081.raw
+raw_data/72377.raw
+raw_data/86938.raw
+raw_data/28992.raw
+raw_data/16366.raw
+raw_data/21919.raw
--- a/app/quantize/data_quantize_person.txt
+++ b/app/quantize/data_quantize_person.txt
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/29.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/21.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/5.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/3.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/24.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/16.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/22.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/13.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/26.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/17.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/35.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/31.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/25.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/Screenshot from 2020-08-24 09-04-49.png
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/6.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/30.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/32.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/37.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/2.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/36.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/28.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/1.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/15.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/crop_26.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/8.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/27.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/9.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/23.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/11.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/34.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/33.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/19.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/10.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/18.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/12.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/20.raw
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/Screenshot from 2020-08-24 09-03-46.png
+/media/ducanh/DATA/tienln/ai_camera/detector/app/quantize/raw_data/processed.jpeg
--- a/app/quantize/raw_data/1.raw
+++ b/app/quantize/raw_data/1.raw
--- a/app/quantize/raw_data/10.raw
+++ b/app/quantize/raw_data/10.raw
--- a/app/quantize/raw_data/11.raw
+++ b/app/quantize/raw_data/11.raw
--- a/app/quantize/raw_data/12.raw
+++ b/app/quantize/raw_data/12.raw
--- a/app/quantize/raw_data/13.raw
+++ b/app/quantize/raw_data/13.raw
--- a/app/quantize/raw_data/15.raw
+++ b/app/quantize/raw_data/15.raw
--- a/app/quantize/raw_data/16.raw
+++ b/app/quantize/raw_data/16.raw
--- a/app/quantize/raw_data/17.raw
+++ b/app/quantize/raw_data/17.raw
--- a/app/quantize/raw_data/18.raw
+++ b/app/quantize/raw_data/18.raw
--- a/app/quantize/raw_data/19.raw
+++ b/app/quantize/raw_data/19.raw
--- a/app/quantize/raw_data/2.raw
+++ b/app/quantize/raw_data/2.raw
--- a/app/quantize/raw_data/20.raw
+++ b/app/quantize/raw_data/20.raw
--- a/app/quantize/raw_data/21.raw
+++ b/app/quantize/raw_data/21.raw
--- a/app/quantize/raw_data/22.raw
+++ b/app/quantize/raw_data/22.raw
--- a/app/quantize/raw_data/23.raw
+++ b/app/quantize/raw_data/23.raw
--- a/app/quantize/raw_data/24.raw
+++ b/app/quantize/raw_data/24.raw
--- a/app/quantize/raw_data/25.raw
+++ b/app/quantize/raw_data/25.raw
--- a/app/quantize/raw_data/26.raw
+++ b/app/quantize/raw_data/26.raw
--- a/app/quantize/raw_data/27.raw
+++ b/app/quantize/raw_data/27.raw
--- a/app/quantize/raw_data/28.raw
+++ b/app/quantize/raw_data/28.raw
--- a/app/quantize/raw_data/29.raw
+++ b/app/quantize/raw_data/29.raw
--- a/app/quantize/raw_data/3.raw
+++ b/app/quantize/raw_data/3.raw
--- a/app/quantize/raw_data/30.raw
+++ b/app/quantize/raw_data/30.raw
--- a/app/quantize/raw_data/31.raw
+++ b/app/quantize/raw_data/31.raw
--- a/app/quantize/raw_data/32.raw
+++ b/app/quantize/raw_data/32.raw
--- a/app/quantize/raw_data/33.raw
+++ b/app/quantize/raw_data/33.raw
--- a/app/quantize/raw_data/34.raw
+++ b/app/quantize/raw_data/34.raw
--- a/app/quantize/raw_data/35.raw
+++ b/app/quantize/raw_data/35.raw
--- a/app/quantize/raw_data/36.raw
+++ b/app/quantize/raw_data/36.raw
--- a/app/quantize/raw_data/37.raw
+++ b/app/quantize/raw_data/37.raw
--- a/app/quantize/raw_data/5.raw
+++ b/app/quantize/raw_data/5.raw
--- a/app/quantize/raw_data/6.raw
+++ b/app/quantize/raw_data/6.raw
--- a/app/quantize/raw_data/8.raw
+++ b/app/quantize/raw_data/8.raw
--- a/app/quantize/raw_data/9.raw
+++ b/app/quantize/raw_data/9.raw
--- a/app/quantize/raw_data/Screenshot from 2020-08-24 09-03-46.png
+++ b/app/quantize/raw_data/Screenshot from 2020-08-24 09-03-46.png
--- a/app/quantize/raw_data/Screenshot from 2020-08-24 09-04-49.png
+++ b/app/quantize/raw_data/Screenshot from 2020-08-24 09-04-49.png
--- a/app/quantize/raw_data/crop_26.raw
+++ b/app/quantize/raw_data/crop_26.raw
--- a/app/quantize/raw_data/processed.jpeg
+++ b/app/quantize/raw_data/processed.jpeg
--- a/app/quantize/raw_data_person/10118.raw
+++ b/app/quantize/raw_data_person/10118.raw
--- a/app/quantize/raw_data_person/12155.raw
+++ b/app/quantize/raw_data_person/12155.raw
--- a/app/quantize/raw_data_person/12384.raw
+++ b/app/quantize/raw_data_person/12384.raw
--- a/app/quantize/raw_data_person/14942.raw
+++ b/app/quantize/raw_data_person/14942.raw
--- a/app/quantize/raw_data_person/15378.raw
+++ b/app/quantize/raw_data_person/15378.raw
--- a/app/quantize/raw_data_person/16366.raw
+++ b/app/quantize/raw_data_person/16366.raw
--- a/app/quantize/raw_data_person/17374.raw
+++ b/app/quantize/raw_data_person/17374.raw
--- a/app/quantize/raw_data_person/18074.raw
+++ b/app/quantize/raw_data_person/18074.raw
--- a/app/quantize/raw_data_person/19030.raw
+++ b/app/quantize/raw_data_person/19030.raw
--- a/app/quantize/raw_data_person/21081.raw
+++ b/app/quantize/raw_data_person/21081.raw
--- a/app/quantize/raw_data_person/21594.raw
+++ b/app/quantize/raw_data_person/21594.raw
--- a/app/quantize/raw_data_person/21919.raw
+++ b/app/quantize/raw_data_person/21919.raw
--- a/app/quantize/raw_data_person/22111.raw
+++ b/app/quantize/raw_data_person/22111.raw
--- a/app/quantize/raw_data_person/22842.raw
+++ b/app/quantize/raw_data_person/22842.raw
--- a/app/quantize/raw_data_person/23324.raw
+++ b/app/quantize/raw_data_person/23324.raw
--- a/app/quantize/raw_data_person/25011.raw
+++ b/app/quantize/raw_data_person/25011.raw
--- a/app/quantize/raw_data_person/25730.raw
+++ b/app/quantize/raw_data_person/25730.raw
--- a/app/quantize/raw_data_person/25929.raw
+++ b/app/quantize/raw_data_person/25929.raw
--- a/app/quantize/raw_data_person/25941.raw
+++ b/app/quantize/raw_data_person/25941.raw
--- a/app/quantize/raw_data_person/26285.raw
+++ b/app/quantize/raw_data_person/26285.raw
--- a/app/quantize/raw_data_person/27457.raw
+++ b/app/quantize/raw_data_person/27457.raw
--- a/app/quantize/raw_data_person/28992.raw
+++ b/app/quantize/raw_data_person/28992.raw
--- a/app/quantize/raw_data_person/29345.raw
+++ b/app/quantize/raw_data_person/29345.raw
--- a/app/quantize/raw_data_person/29547.raw
+++ b/app/quantize/raw_data_person/29547.raw
--- a/app/quantize/raw_data_person/32689.raw
+++ b/app/quantize/raw_data_person/32689.raw
--- a/app/quantize/raw_data_person/34577.raw
+++ b/app/quantize/raw_data_person/34577.raw
--- a/app/quantize/raw_data_person/37103.raw
+++ b/app/quantize/raw_data_person/37103.raw
--- a/app/quantize/raw_data_person/37352.raw
+++ b/app/quantize/raw_data_person/37352.raw
--- a/app/quantize/raw_data_person/37585.raw
+++ b/app/quantize/raw_data_person/37585.raw
--- a/app/quantize/raw_data_person/39834.raw
+++ b/app/quantize/raw_data_person/39834.raw
--- a/app/quantize/raw_data_person/42194.raw
+++ b/app/quantize/raw_data_person/42194.raw
--- a/app/quantize/raw_data_person/42987.raw
+++ b/app/quantize/raw_data_person/42987.raw
--- a/app/quantize/raw_data_person/43464.raw
+++ b/app/quantize/raw_data_person/43464.raw
--- a/app/quantize/raw_data_person/44600.raw
+++ b/app/quantize/raw_data_person/44600.raw
--- a/app/quantize/raw_data_person/44684.raw
+++ b/app/quantize/raw_data_person/44684.raw
--- a/app/quantize/raw_data_person/46879.raw
+++ b/app/quantize/raw_data_person/46879.raw
--- a/app/quantize/raw_data_person/46905.raw
+++ b/app/quantize/raw_data_person/46905.raw
--- a/app/quantize/raw_data_person/47457.raw
+++ b/app/quantize/raw_data_person/47457.raw
--- a/app/quantize/raw_data_person/47692.raw
+++ b/app/quantize/raw_data_person/47692.raw
--- a/app/quantize/raw_data_person/49723.raw
+++ b/app/quantize/raw_data_person/49723.raw
--- a/app/quantize/raw_data_person/50707.raw
+++ b/app/quantize/raw_data_person/50707.raw
--- a/app/quantize/raw_data_person/51264.raw
+++ b/app/quantize/raw_data_person/51264.raw
--- a/app/quantize/raw_data_person/51772.raw
+++ b/app/quantize/raw_data_person/51772.raw
--- a/app/quantize/raw_data_person/51994.raw
+++ b/app/quantize/raw_data_person/51994.raw
--- a/app/quantize/raw_data_person/52162.raw
+++ b/app/quantize/raw_data_person/52162.raw
--- a/app/quantize/raw_data_person/52309.raw
+++ b/app/quantize/raw_data_person/52309.raw
--- a/app/quantize/raw_data_person/53293.raw
+++ b/app/quantize/raw_data_person/53293.raw
--- a/app/quantize/raw_data_person/53785.raw
+++ b/app/quantize/raw_data_person/53785.raw
--- a/app/quantize/raw_data_person/55038.raw
+++ b/app/quantize/raw_data_person/55038.raw
--- a/app/quantize/raw_data_person/56403.raw
+++ b/app/quantize/raw_data_person/56403.raw
--- a/app/quantize/raw_data_person/58027.raw
+++ b/app/quantize/raw_data_person/58027.raw
--- a/app/quantize/raw_data_person/58802.raw
+++ b/app/quantize/raw_data_person/58802.raw
--- a/app/quantize/raw_data_person/59368.raw
+++ b/app/quantize/raw_data_person/59368.raw
--- a/app/quantize/raw_data_person/59429.raw
+++ b/app/quantize/raw_data_person/59429.raw
--- a/app/quantize/raw_data_person/61017.raw
+++ b/app/quantize/raw_data_person/61017.raw
--- a/app/quantize/raw_data_person/61103.raw
+++ b/app/quantize/raw_data_person/61103.raw
--- a/app/quantize/raw_data_person/63513.raw
+++ b/app/quantize/raw_data_person/63513.raw
--- a/app/quantize/raw_data_person/63752.raw
+++ b/app/quantize/raw_data_person/63752.raw
--- a/app/quantize/raw_data_person/66063.raw
+++ b/app/quantize/raw_data_person/66063.raw
--- a/app/quantize/raw_data_person/68275.raw
+++ b/app/quantize/raw_data_person/68275.raw
--- a/app/quantize/raw_data_person/68371.raw
+++ b/app/quantize/raw_data_person/68371.raw
--- a/app/quantize/raw_data_person/69533.raw
+++ b/app/quantize/raw_data_person/69533.raw
--- a/app/quantize/raw_data_person/69667.raw
+++ b/app/quantize/raw_data_person/69667.raw
--- a/app/quantize/raw_data_person/70275.raw
+++ b/app/quantize/raw_data_person/70275.raw
--- a/app/quantize/raw_data_person/70829.raw
+++ b/app/quantize/raw_data_person/70829.raw
--- a/app/quantize/raw_data_person/72055.raw
+++ b/app/quantize/raw_data_person/72055.raw
--- a/app/quantize/raw_data_person/72073.raw
+++ b/app/quantize/raw_data_person/72073.raw
--- a/app/quantize/raw_data_person/72377.raw
+++ b/app/quantize/raw_data_person/72377.raw
--- a/app/quantize/raw_data_person/72880.raw
+++ b/app/quantize/raw_data_person/72880.raw
--- a/app/quantize/raw_data_person/75792.raw
+++ b/app/quantize/raw_data_person/75792.raw
--- a/app/quantize/raw_data_person/77275.raw
+++ b/app/quantize/raw_data_person/77275.raw
--- a/app/quantize/raw_data_person/78356.raw
+++ b/app/quantize/raw_data_person/78356.raw
--- a/app/quantize/raw_data_person/78851.raw
+++ b/app/quantize/raw_data_person/78851.raw
--- a/app/quantize/raw_data_person/79228.raw
+++ b/app/quantize/raw_data_person/79228.raw
--- a/app/quantize/raw_data_person/82567.raw
+++ b/app/quantize/raw_data_person/82567.raw
--- a/app/quantize/raw_data_person/84186.raw
+++ b/app/quantize/raw_data_person/84186.raw
--- a/app/quantize/raw_data_person/85954.raw
+++ b/app/quantize/raw_data_person/85954.raw
--- a/app/quantize/raw_data_person/86928.raw
+++ b/app/quantize/raw_data_person/86928.raw
--- a/app/quantize/raw_data_person/86938.raw
+++ b/app/quantize/raw_data_person/86938.raw
--- a/datasets/__pycache__/data_loader.cpython-36.pyc
+++ b/datasets/__pycache__/data_loader.cpython-36.pyc
--- a/datasets/__pycache__/data_preprocessing.cpython-36.pyc
+++ b/datasets/__pycache__/data_preprocessing.cpython-36.pyc
--- a/datasets/__pycache__/data_preprocessing.cpython-37.pyc
+++ b/datasets/__pycache__/data_preprocessing.cpython-37.pyc
--- a/datasets/__pycache__/data_transform.cpython-36.pyc
+++ b/datasets/__pycache__/data_transform.cpython-36.pyc
--- a/datasets/__pycache__/data_transform.cpython-37.pyc
+++ b/datasets/__pycache__/data_transform.cpython-37.pyc
--- a/datasets/data_loader.py
+++ b/datasets/data_loader.py
--- a/datasets/data_preprocessing.py
+++ b/datasets/data_preprocessing.py
--- a/datasets/data_transform.py
+++ b/datasets/data_transform.py
--- a/datasets/train_dataset.txt
+++ b/datasets/train_dataset.txt
--- a/datasets/valid_dataset.txt
+++ b/datasets/valid_dataset.txt
--- a/docs/head.md
+++ b/docs/head.md
--- a/docs/head.pdf
+++ b/docs/head.pdf
--- a/docs/person.md
+++ b/docs/person.md
--- a/docs/person.pdf
+++ b/docs/person.pdf
--- a/model/__pycache__/mb_ssd_lite.cpython-36.pyc
+++ b/model/__pycache__/mb_ssd_lite.cpython-36.pyc
--- a/model/__pycache__/mb_ssd_lite_f19.cpython-36.pyc
+++ b/model/__pycache__/mb_ssd_lite_f19.cpython-36.pyc
--- a/model/__pycache__/mb_ssd_lite_f19.cpython-37.pyc
+++ b/model/__pycache__/mb_ssd_lite_f19.cpython-37.pyc
--- a/model/__pycache__/mb_ssd_lite_f38.cpython-36.pyc
+++ b/model/__pycache__/mb_ssd_lite_f38.cpython-36.pyc
--- a/model/__pycache__/mb_ssd_lite_f38.cpython-37.pyc
+++ b/model/__pycache__/mb_ssd_lite_f38.cpython-37.pyc
--- a/model/__pycache__/mb_ssd_lite_f38_person.cpython-36.pyc
+++ b/model/__pycache__/mb_ssd_lite_f38_person.cpython-36.pyc
--- a/model/__pycache__/mb_ssd_lite_f38_person.cpython-37.pyc
+++ b/model/__pycache__/mb_ssd_lite_f38_person.cpython-37.pyc
--- a/model/__pycache__/rfb_tiny_mb_ssd.cpython-36.pyc
+++ b/model/__pycache__/rfb_tiny_mb_ssd.cpython-36.pyc
--- a/model/__pycache__/rfb_tiny_mb_ssd.cpython-37.pyc
+++ b/model/__pycache__/rfb_tiny_mb_ssd.cpython-37.pyc
--- a/model/config/__pycache__/mb_ssd_lite_f19_config.cpython-36.pyc
+++ b/model/config/__pycache__/mb_ssd_lite_f19_config.cpython-36.pyc
--- a/model/config/__pycache__/mb_ssd_lite_f19_config.cpython-37.pyc
+++ b/model/config/__pycache__/mb_ssd_lite_f19_config.cpython-37.pyc
--- a/model/config/__pycache__/mb_ssd_lite_f38_config.cpython-36.pyc
+++ b/model/config/__pycache__/mb_ssd_lite_f38_config.cpython-36.pyc
--- a/model/config/__pycache__/mb_ssd_lite_f38_config.cpython-37.pyc
+++ b/model/config/__pycache__/mb_ssd_lite_f38_config.cpython-37.pyc
--- a/model/config/__pycache__/mb_ssd_lite_f38_person_config.cpython-36.pyc
+++ b/model/config/__pycache__/mb_ssd_lite_f38_person_config.cpython-36.pyc
--- a/model/config/__pycache__/mb_ssd_lite_f38_person_config.cpython-37.pyc
+++ b/model/config/__pycache__/mb_ssd_lite_f38_person_config.cpython-37.pyc
--- a/model/config/__pycache__/mobilenetv1_ssd_config.cpython-36.pyc
+++ b/model/config/__pycache__/mobilenetv1_ssd_config.cpython-36.pyc
--- a/model/config/__pycache__/rfb_tiny_mb_ssd_config.cpython-36.pyc
+++ b/model/config/__pycache__/rfb_tiny_mb_ssd_config.cpython-36.pyc
--- a/model/config/__pycache__/rfb_tiny_mb_ssd_config.cpython-37.pyc
+++ b/model/config/__pycache__/rfb_tiny_mb_ssd_config.cpython-37.pyc
--- a/model/config/mb_ssd_lite_f19_config.py
+++ b/model/config/mb_ssd_lite_f19_config.py
--- a/model/config/mb_ssd_lite_f38_config.py
+++ b/model/config/mb_ssd_lite_f38_config.py
--- a/model/config/mb_ssd_lite_f38_person_config.py
+++ b/model/config/mb_ssd_lite_f38_person_config.py
--- a/model/config/rfb_tiny_mb_ssd_config.py
+++ b/model/config/rfb_tiny_mb_ssd_config.py
--- a/model/mb_ssd_lite_f19.py
+++ b/model/mb_ssd_lite_f19.py
--- a/model/mb_ssd_lite_f38.py
+++ b/model/mb_ssd_lite_f38.py
--- a/model/mb_ssd_lite_f38_person.py
+++ b/model/mb_ssd_lite_f38_person.py
--- a/model/rfb_tiny_mb_ssd.py
+++ b/model/rfb_tiny_mb_ssd.py
--- a/module/__pycache__/mobilent_v2.cpython-36.pyc
+++ b/module/__pycache__/mobilent_v2.cpython-36.pyc
--- a/module/__pycache__/mobilent_v2.cpython-37.pyc
+++ b/module/__pycache__/mobilent_v2.cpython-37.pyc
--- a/module/__pycache__/rfb_tiny_mobilenet_v2.cpython-36.pyc
+++ b/module/__pycache__/rfb_tiny_mobilenet_v2.cpython-36.pyc
--- a/module/__pycache__/rfb_tiny_mobilenet_v2.cpython-37.pyc
+++ b/module/__pycache__/rfb_tiny_mobilenet_v2.cpython-37.pyc
--- a/module/__pycache__/ssd.cpython-36.pyc
+++ b/module/__pycache__/ssd.cpython-36.pyc
--- a/module/__pycache__/ssd.cpython-37.pyc
+++ b/module/__pycache__/ssd.cpython-37.pyc
--- a/module/mobilent_v2.py
+++ b/module/mobilent_v2.py
--- a/module/rfb_tiny_mobilenet_v2.py
+++ b/module/rfb_tiny_mobilenet_v2.py
--- a/module/ssd.py
+++ b/module/ssd.py
--- a/pre-train/mb2-imagenet-71_8.pth
+++ b/pre-train/mb2-imagenet-71_8.pth
--- a/pre-train/mb2-ssd-lite-mp-0_686.pth
+++ b/pre-train/mb2-ssd-lite-mp-0_686.pth
--- a/test/detect_imgs.py
+++ b/test/detect_imgs.py
--- a/test/live_demo.py
+++ b/test/live_demo.py
--- a/train/__init__.py
+++ b/train/__init__.py
--- a/train/__pycache__/train.cpython-36.pyc
+++ b/train/__pycache__/train.cpython-36.pyc
--- a/train/main.py
+++ b/train/main.py
--- a/train/train.py
+++ b/train/train.py
--- a/train/weight/training/mb2-ssd-lite_f19/mb2-ssd-lite_f19-epoch-0-train_loss-3.19-val_loss-3.53.pth
+++ b/train/weight/training/mb2-ssd-lite_f19/mb2-ssd-lite_f19-epoch-0-train_loss-3.19-val_loss-3.53.pth
--- a/train/weight/training/mb2-ssd-lite_f19/mb2-ssd-lite_f19-epoch-0-train_loss-4.29-val_loss-5.88.pth
+++ b/train/weight/training/mb2-ssd-lite_f19/mb2-ssd-lite_f19-epoch-0-train_loss-4.29-val_loss-5.88.pth
--- a/train/weight/training/mb2-ssd-lite_f19/mb2-ssd-lite_f19-epoch-1-train_loss-2.98-val_loss-3.24.pth
+++ b/train/weight/training/mb2-ssd-lite_f19/mb2-ssd-lite_f19-epoch-1-train_loss-2.98-val_loss-3.24.pth
--- a/train/weight/training/mb2-ssd-lite_f19/mb2-ssd-lite_f19-epoch-2-train_loss-2.75-val_loss-3.01.pth
+++ b/train/weight/training/mb2-ssd-lite_f19/mb2-ssd-lite_f19-epoch-2-train_loss-2.75-val_loss-3.01.pth
--- a/train/weight/training/mb2-ssd-lite_f19/mb2-ssd-lite_f19-epoch-3-train_loss-2.6-val_loss-2.84.pth
+++ b/train/weight/training/mb2-ssd-lite_f19/mb2-ssd-lite_f19-epoch-3-train_loss-2.6-val_loss-2.84.pth
--- a/train/weight/training/mb2-ssd-lite_f19/mb2-ssd-lite_f19-epoch-4-train_loss-2.49-val_loss-2.86.pth
+++ b/train/weight/training/mb2-ssd-lite_f19/mb2-ssd-lite_f19-epoch-4-train_loss-2.49-val_loss-2.86.pth
--- a/train/weight/training/mb2-ssd-lite_f19/mb2-ssd-lite_f19-epoch-5-train_loss-2.43-val_loss-2.72.pth
+++ b/train/weight/training/mb2-ssd-lite_f19/mb2-ssd-lite_f19-epoch-5-train_loss-2.43-val_loss-2.72.pth
--- a/train/weight/training/mb2-ssd-lite_f19/mb2-ssd-lite_f19-epoch-6-train_loss-2.42-val_loss-2.68.pth
+++ b/train/weight/training/mb2-ssd-lite_f19/mb2-ssd-lite_f19-epoch-6-train_loss-2.42-val_loss-2.68.pth
--- a/utils/__init__.py
+++ b/utils/__init__.py
--- a/utils/__pycache__/__init__.cpython-36.pyc
+++ b/utils/__pycache__/__init__.cpython-36.pyc
--- a/utils/__pycache__/__init__.cpython-37.pyc
+++ b/utils/__pycache__/__init__.cpython-37.pyc
--- a/utils/__pycache__/argument.cpython-36.pyc
+++ b/utils/__pycache__/argument.cpython-36.pyc
--- a/utils/__pycache__/argument.cpython-37.pyc
+++ b/utils/__pycache__/argument.cpython-37.pyc
--- a/utils/__pycache__/box_processing.cpython-36.pyc
+++ b/utils/__pycache__/box_processing.cpython-36.pyc
--- a/utils/__pycache__/box_processing.cpython-37.pyc
+++ b/utils/__pycache__/box_processing.cpython-37.pyc
--- a/utils/__pycache__/loss.cpython-36.pyc
+++ b/utils/__pycache__/loss.cpython-36.pyc
--- a/utils/__pycache__/misc.cpython-36.pyc
+++ b/utils/__pycache__/misc.cpython-36.pyc
--- a/utils/__pycache__/misc.cpython-37.pyc
+++ b/utils/__pycache__/misc.cpython-37.pyc
--- a/utils/__pycache__/predictor.cpython-36.pyc
+++ b/utils/__pycache__/predictor.cpython-36.pyc
--- a/utils/__pycache__/predictor.cpython-37.pyc
+++ b/utils/__pycache__/predictor.cpython-37.pyc
--- a/utils/argument.py
+++ b/utils/argument.py
--- a/utils/box_processing.py
+++ b/utils/box_processing.py
--- a/utils/l2_norm.py
+++ b/utils/l2_norm.py
--- a/utils/labels/head.txt
+++ b/utils/labels/head.txt
--- a/utils/labels/person.txt
+++ b/utils/labels/person.txt
--- a/utils/loss.py
+++ b/utils/loss.py
--- a/utils/misc.py
+++ b/utils/misc.py
--- a/utils/predictor.py
+++ b/utils/predictor.py
--- a/utils/pth2onnx.py
+++ b/utils/pth2onnx.py