This project aims to simplify objection detection. You can do all your work in a Jupyter Notebook rather than in command line. You can choose Yolo or Retinanet or both to train you model.
This is the tutorial video in Chinese for your reference.
Please read carefully, and conform the criteria below.
Your Dataset folder MUST contain a folder named VOCdevkit, which includes your datafoler such as VOC2007, Helmet(The specific object your want to detect). The treeview of that folder looks like this. If you are confused with it, you can refer to the Pascal VOC dataset.
├─Annotations
│ └─*.xml
├─ImageSets
│ └─Main
│ | └─ <classname>_train.txt,<classname>_val.txt, train.txt, val.txt
└─JPEGImages
│ └─*.jpg(lower case extension name. MUST be end with jpg rather than jpeg)
If you don't hava train.txt and val.txt in Main folder, copy Gentxt.py
to your dataset's ImageSets
folder, and run that script, then you will get these text files.
eg. If I want to train one object detection on helmet, my file tree can be like this image.
Make sure everything is well prepared, compress the VOCdevkit folder to a zip/tar file.
Choose one way below to deploy, these two ways both work. In the web page you need input some info to deploy a VM, you can type you own password but DO NOT change the username.
eg. If you click the first button to deploy, just fill in the Visual Machine Name and Admin Password cells. If you want use your existing resource group, check the selection box below in the yellow rectangle.
note. In some resource group or vm location, you may not access the url in the following step, because the custom script failed. If it happends, try eastus2.
After deployment, you can get an ip address in Azure Portal, open the url http://<ip>:9999
in your browser. Login with password msfthack
and you will see Yolo and Retinanet notebooks, choose one to open and follow the steps.
And if you stop/restart the vm, next time you should login it through SSH, and run jupyter notebook
in command line, when it's ready, visit the 9999 port.
In the webpage you will see:
You should upload your dateset file in the Image folder, in the picture I uploaded two datesets. If you unzip the datasests, you will see the VOCdevkit folder.
In the VOCdevkit folder, you can see all your datasets' folder.