Skip to content

alanhaugen/TMAT3004-Bacheloroppgave

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

TMAT3004-Bacheloroppgave

trainer is based on RetinaNet and detectron2 https://detectron2.readthedocs.io/tutorials/install.html#common-installation-issues

App is written in C++. I recommend QtCreator open-source, and msvc++ from Microsoft Visual Studio Community Edition.

QtCreator can open the cmake file CMakeLists.txt

You need OpenCV, see https://medium.com/beesightsoft/build-opencv-opencv-contrib-on-windows-2e3b1ca96955

Please use OpenCV 3.4, it has YOLOv4 support.

Collect your own dataset and put it into this directory, call the folder data. The structure is described in the report.

Project overview

app consists of several projects

OBJECT_DETECTOR is the important project, it loads the model found in the data to count cod and saithe.

DATASET_TOOL is for improving the dataset, to create new labels

See Releases for executables of the project. You will find Windows and MacOS binaries. The Windows binaries include pre-compiled OpenCV libraries.

RetinaNet

train.py will train a model with RetinaNet

inference.py will create a video with the RetinaNet model

Report

report consists of my bachelor thesis

About

๐ŸŸ๐Ÿ” Detect Atlantic cod and saithe in video, bachelor thesis

Resources

Stars

Watchers

Forks

Packages

No packages published