• 设为首页
  • 点击收藏
  • 手机版
    手机扫一扫访问
    迪恩网络手机版
  • 关注官方公众号
    微信扫一扫关注
    迪恩网络公众号

DrewNF/Tensorflow_Object_Tracking_Video: Object Tracking in Tensorflow ( Localiz ...

原作者: [db:作者] 来自: 网络 收藏 邀请

开源软件名称(OpenSource Name):

DrewNF/Tensorflow_Object_Tracking_Video

开源软件地址(OpenSource Url):

https://github.com/DrewNF/Tensorflow_Object_Tracking_Video

开源编程语言(OpenSource Language):

Python 91.0%

开源软件介绍(OpenSource Introduction):

Tensorflow_Object_Tracking_Video

(Version 0.3, Last Update 10-03-2017)

alt text alt text alt text alt text

The Project follow the below index:

  1. Introduction;
  2. Requitements & Installation;
  3. YOLO Script Usage
    1. Setting Parameters;
    2. Usage.
  4. VID TENSORBOX Script Usage
    1. Setting Parameters;
    2. Usage.
  5. TENSORBOX Tests Files;
  6. Dataset Scripts;
  7. Copyright;
  8. State of the Project.
  9. DOWNLOADS.
  10. Acknowledgements.
  11. Bibliography.

1.Introduction

This Repository is my Master Thesis Project: "Develop a Video Object Tracking with Tensorflow Technology" and it's still developing, so many updates will be made. In this work, I used the architecture and problem solving strategy of the Paper T-CNN(Arxiv), that won last year IMAGENET 2015 Teaser Challenge VID. So the whole script architecture will be made of several component in cascade:

  1. Still Image Detection (Return Tracking Results on single Frame);
  2. Temporal Information Detection( Introducing Temporal Information into the DET Results);
  3. Context Information Detection( Introducing Context Information into the DET Results);

Notice that the Still Image Detection component could be unique or decompose into two sub-component:

  1. First: determinate "Where" in the Frame;
  2. Second: determinate "What" in the Frame.

My project use many online tensorflow projects, as:

2.Requirement & Installation

To install the script you only need to download the Repository. To Run the script you have to had installed:

  • Tensorflow;
  • OpenCV;
  • Python;

All the Python library necessary could be installed easily trought pip install package-name. If you want to follow a guide to install the requirements here is the link for a tutorial I wrote for myself and for a course of Deep Learning at UPC.

3.YOLO Script Usage

You only look once (YOLO) is a state-of-the-art, real-time object detection system.## i.Setting Parameters This are the inline terminal argmunts taken from the script, most of them aren't required, only the video path must be specified when we call the script:

  parser = argparse.ArgumentParser()
  parser.add_argument('--det_frames_folder', default='det_frames/', type=str)
  parser.add_argument('--det_result_folder', default='det_results/', type=str)
  parser.add_argument('--result_folder', default='summary_result/', type=str)
  parser.add_argument('--summary_file', default='results.txt', type=str)
  parser.add_argument('--output_name', default='output.mp4', type=str)
  parser.add_argument('--perc', default=5, type=int)
  parser.add_argument('--path_video', required=True, type=str)

Now you have to download the weights for YOLO and put them into /YOLO_DET_Alg/weights/.

For YOLO knowledge here you can find Original code(C implementation) & paper.

ii.Usage

After Set the Parameters, we can proceed and run the script:

  python VID_yolo.py --path_video video.mp4

You will see some Terminal Output like:

alt tag

You will see a realtime frames output(like the one here below) and then finally all will be embedded into the Video Output( I uploaded the first two Test I've made in the folder /video_result, you can download them and take a look to the final result. The first one has problems in the frames order, this is why you will see so much flickering in the video image,the problem was then solved and in the second doesn't show frames flickering ):

alt tag

4.VID TENSORBOX Script Usage

i.Setting Parameters

This are the inline terminal argmunts taken from the script, most of them aren't required. As before, only the video path must be specified when we call the script:

  parser.add_argument('--output_name', default='output.mp4', type=str)
  parser.add_argument('--hypes', default='./hypes/overfeat_rezoom.json', type=str)
  parser.add_argument('--weights', default='./output/save.ckpt-1090000', type=str)
  parser.add_argument('--perc', default=2, type=int)
  parser.add_argument('--path_video', required=True, type=str)

I will soon put a weight file to download. For train and spec on the multiclass implementation I will add them after the end of my thesis project.

ii.Usage

Download the .zip files linked in the Download section and replace the folders.

Then, after set the parameters, we can proceed and run the script:

  python VID_tensorbox_multi_class.py --path_video video.mp4

5.Tensorbox Tests

In the folder video_result_OVT you can find files result of the runs of the VID TENSOBOX scripts.

6.Dataset Scripts

All the scripts below are for the VID classes so if you wonna adapt them for other you have to simply change the Classes.py file where are defined the correspondencies between codes and names. All the data on the image are made respect a specific Image Ratio, because TENSORBOX works only with 640x480 PNG images, you will have to change the code a little to adapt to your needs. I will provide four scripts:

  1. Process_Dataset_heavy.py: Process your dataset with a brute force approach, you will obtain more bbox and files for each class;
  2. Process_Dataset_lightweight.py: Process your dataset with a lightweight approach making, you will obtain less bbox and files for each class;
  3. Resize_Dataset.py: Resize your dataset to 640x480 PNG images;
  4. Test_Processed_Data.py: Will test that the process end well without errors.

I've also add some file scripts to pre process and prepare the dataset to train the last component, the Inception Model, you can find them in a subfolder of the dataset script folder.

7.Copyright

According to the LICENSE file of the original code,

  • Me and original author hold no liability for any damages;
  • Do not use this on commercial!.

8.State of the Project

  • Support YOLO (SingleClass) DET Algorithm;
  • Support Training ONLY TENSOBOX and INCEPTION Training;
  • USE OF TEMPORAL INFORMATION [This are retrieved through some post processing algorithm I've implemented in the Utils_Video.py file NOT TRAINABLE];
  • Modular Architecture composed in cascade by: Tensorbox (as General Object Detector), Tracker and Smoother and Inception (as Object Classifier);

9.Downloads

Here below the links of the weights file for Inception and Tensorbox from my retraining experiments:

10.Acknowledgements

Thanks to Professors:

  • Elena Baralis from Politecnico di Torino Dipartimento di Automatica e Informatica;
  • Jordi Torres from BSC Department of Computer Science;
  • Xavi Giro ”I” Nieto from UPC Department of Image Processing.

11.Bibliography

i.Course

ii.Classification

iii.Detection

iv.Tracking




鲜花

握手

雷人

路过

鸡蛋
该文章已有0人参与评论

请发表评论

全部评论

专题导读
热门推荐
阅读排行榜

扫描微信二维码

查看手机版网站

随时了解更新最新资讯

139-2527-9053

在线客服(服务时间 9:00~18:00)

在线QQ客服
地址:深圳市南山区西丽大学城创智工业园
电邮:jeky_zhao#qq.com
移动电话:139-2527-9053

Powered by 互联科技 X3.4© 2001-2213 极客世界.|Sitemap