diff --git a/.idea/deepmot.iml b/.idea/deepmot.iml deleted file mode 100644 index 7c9d48f0fdb082ceb3555fd1dc94cc6e152b70a6..0000000000000000000000000000000000000000 --- a/.idea/deepmot.iml +++ /dev/null @@ -1,12 +0,0 @@ -<?xml version="1.0" encoding="UTF-8"?> -<module type="PYTHON_MODULE" version="4"> - <component name="NewModuleRootManager"> - <content url="file://$MODULE_DIR$" /> - <orderEntry type="inheritedJdk" /> - <orderEntry type="sourceFolder" forTests="false" /> - </component> - <component name="TestRunnerService"> - <option name="projectConfiguration" value="pytest" /> - <option name="PROJECT_TEST_RUNNER" value="pytest" /> - </component> -</module> \ No newline at end of file diff --git a/.idea/misc.xml b/.idea/misc.xml deleted file mode 100644 index 65531ca992813bbfedbe43dfae5a5f4337168ed8..0000000000000000000000000000000000000000 --- a/.idea/misc.xml +++ /dev/null @@ -1,4 +0,0 @@ -<?xml version="1.0" encoding="UTF-8"?> -<project version="4"> - <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.6" project-jdk-type="Python SDK" /> -</project> \ No newline at end of file diff --git a/.idea/modules.xml b/.idea/modules.xml deleted file mode 100644 index cda52c832f2f537abfd6bbe4bd33640f922afb50..0000000000000000000000000000000000000000 --- a/.idea/modules.xml +++ /dev/null @@ -1,8 +0,0 @@ -<?xml version="1.0" encoding="UTF-8"?> -<project version="4"> - <component name="ProjectModuleManager"> - <modules> - <module fileurl="file://$PROJECT_DIR$/.idea/deepmot.iml" filepath="$PROJECT_DIR$/.idea/deepmot.iml" /> - </modules> - </component> -</project> \ No newline at end of file diff --git a/.idea/vcs.xml b/.idea/vcs.xml deleted file mode 100644 index 94a25f7f4cb416c083d265558da75d457237d671..0000000000000000000000000000000000000000 --- a/.idea/vcs.xml +++ /dev/null @@ -1,6 +0,0 @@ -<?xml version="1.0" encoding="UTF-8"?> -<project version="4"> - <component name="VcsDirectoryMappings"> - <mapping directory="$PROJECT_DIR$" vcs="Git" /> - </component> -</project> \ No newline at end of file diff --git a/.idea/workspace.xml b/.idea/workspace.xml deleted file mode 100644 index 671792ffbaf16ddcb87dde95b749ea08b4f68606..0000000000000000000000000000000000000000 --- a/.idea/workspace.xml +++ /dev/null @@ -1,288 +0,0 @@ -<?xml version="1.0" encoding="UTF-8"?> -<project version="4"> - <component name="ChangeListManager"> - <list default="true" id="0fadd8fa-3a59-465d-b730-df94709c4cb7" name="Default Changelist" comment="update readme."> - <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" /> - <change beforePath="$PROJECT_DIR$/README.md" beforeDir="false" afterPath="$PROJECT_DIR$/README.md" afterDir="false" /> - <change beforePath="$PROJECT_DIR$/tracking_on_mot.py" beforeDir="false" afterPath="$PROJECT_DIR$/tracking_on_mot.py" afterDir="false" /> - </list> - <option name="EXCLUDED_CONVERTED_TO_IGNORED" value="true" /> - <option name="SHOW_DIALOG" value="false" /> - <option name="HIGHLIGHT_CONFLICTS" value="true" /> - <option name="HIGHLIGHT_NON_ACTIVE_CHANGELIST" value="false" /> - <option name="LAST_RESOLUTION" value="IGNORE" /> - </component> - <component name="FileEditorManager"> - <leaf SIDE_TABS_SIZE_LIMIT_KEY="300"> - <file pinned="false" current-in-tab="false"> - <entry file="file://$PROJECT_DIR$/evalation.py"> - <provider selected="true" editor-type-id="text-editor"> - <state relative-caret-position="2500"> - <caret line="132" column="69" selection-start-line="132" selection-start-column="69" selection-end-line="132" selection-end-column="69" /> - <folding> - <element signature="e#424#433#0" expanded="true" /> - </folding> - </state> - </provider> - </entry> - </file> - <file pinned="false" current-in-tab="true"> - <entry file="file://$PROJECT_DIR$/tracking_on_mot.py"> - <provider selected="true" editor-type-id="text-editor"> - <state relative-caret-position="63"> - <caret line="356" lean-forward="true" selection-start-line="356" selection-end-line="356" /> - <folding> - <element signature="e#423#432#0" expanded="true" /> - </folding> - </state> - </provider> - </entry> - </file> - <file pinned="false" current-in-tab="false"> - <entry file="file://$PROJECT_DIR$/utils/tracking_config.py"> - <provider selected="true" editor-type-id="text-editor"> - <state relative-caret-position="240"> - <caret line="12" column="4" selection-start-line="12" selection-start-column="4" selection-end-line="12" selection-end-column="4" /> - </state> - </provider> - </entry> - </file> - <file pinned="false" current-in-tab="false"> - <entry file="file://$PROJECT_DIR$/train_mot.py"> - <provider selected="true" editor-type-id="text-editor"> - <state relative-caret-position="294"> - <caret line="283" column="44" selection-start-line="283" selection-start-column="44" selection-end-line="283" selection-end-column="44" /> - </state> - </provider> - </entry> - </file> - <file pinned="false" current-in-tab="false"> - <entry file="file://$PROJECT_DIR$/README.md"> - <provider selected="true" editor-type-id="split-provider[text-editor;markdown-preview-editor]"> - <state split_layout="SPLIT"> - <first_editor relative-caret-position="120"> - <caret line="126" column="3" selection-start-line="126" selection-start-column="3" selection-end-line="126" selection-end-column="3" /> - </first_editor> - <second_editor /> - </state> - </provider> - </entry> - </file> - <file pinned="false" current-in-tab="false"> - <entry file="file://$PROJECT_DIR$/plot_results.py"> - <provider selected="true" editor-type-id="text-editor" /> - </entry> - </file> - </leaf> - </component> - <component name="FindInProjectRecents"> - <findStrings> - <find>siamr</find> - </findStrings> - </component> - <component name="Git.Settings"> - <option name="RECENT_GIT_ROOT_PATH" value="$PROJECT_DIR$" /> - </component> - <component name="IdeDocumentHistory"> - <option name="CHANGED_PATHS"> - <list> - <option value="$PROJECT_DIR$/evalation.py" /> - <option value="$PROJECT_DIR$/requirements" /> - <option value="$PROJECT_DIR$/train_mot.py" /> - <option value="$PROJECT_DIR$/SingularityImages/README.md" /> - <option value="$PROJECT_DIR$/README.md" /> - <option value="$PROJECT_DIR$/tracking_on_mot.py" /> - </list> - </option> - </component> - <component name="ProjectFrameBounds" extendedState="6"> - <option name="x" value="67" /> - <option name="y" value="249" /> - <option name="width" value="1843" /> - <option name="height" value="831" /> - </component> - <component name="ProjectView"> - <navigator proportions="" version="1"> - <foldersAlwaysOnTop value="true" /> - </navigator> - <panes> - <pane id="ProjectPane"> - <subPane> - <expand> - <path> - <item name="deepmot" type="b2602c69:ProjectViewProjectNode" /> - <item name="deepmot" type="462c0819:PsiDirectoryNode" /> - </path> - </expand> - <select /> - </subPane> - </pane> - <pane id="Scope" /> - </panes> - </component> - <component name="PropertiesComponent"> - <property name="SHARE_PROJECT_CONFIGURATION_FILES" value="true" /> - <property name="last_opened_file_path" value="$PROJECT_DIR$/SingularityImages" /> - </component> - <component name="RecentsManager"> - <key name="CopyFile.RECENT_KEYS"> - <recent name="$PROJECT_DIR$/SingularityImages" /> - </key> - </component> - <component name="RunDashboard"> - <option name="ruleStates"> - <list> - <RuleState> - <option name="name" value="ConfigurationTypeDashboardGroupingRule" /> - </RuleState> - <RuleState> - <option name="name" value="StatusDashboardGroupingRule" /> - </RuleState> - </list> - </option> - </component> - <component name="SvnConfiguration"> - <configuration /> - </component> - <component name="TaskManager"> - <task active="true" id="Default" summary="Default task"> - <changelist id="0fadd8fa-3a59-465d-b730-df94709c4cb7" name="Default Changelist" comment="" /> - <created>1559933365142</created> - <option name="number" value="Default" /> - <option name="presentableId" value="Default" /> - <updated>1559933365142</updated> - </task> - <task id="LOCAL-00001" summary="correct txt encoding problem."> - <created>1560242415941</created> - <option name="number" value="00001" /> - <option name="presentableId" value="LOCAL-00001" /> - <option name="project" value="LOCAL" /> - <updated>1560242415941</updated> - </task> - <task id="LOCAL-00002" summary="update readme."> - <created>1560243751519</created> - <option name="number" value="00002" /> - <option name="presentableId" value="LOCAL-00002" /> - <option name="project" value="LOCAL" /> - <updated>1560243751519</updated> - </task> - <task id="LOCAL-00003" summary="add Singularity Images Folder"> - <created>1560244485226</created> - <option name="number" value="00003" /> - <option name="presentableId" value="LOCAL-00003" /> - <option name="project" value="LOCAL" /> - <updated>1560244485226</updated> - </task> - <task id="LOCAL-00004" summary="update readme"> - <created>1560273133268</created> - <option name="number" value="00004" /> - <option name="presentableId" value="LOCAL-00004" /> - <option name="project" value="LOCAL" /> - <updated>1560273133268</updated> - </task> - <task id="LOCAL-00005" summary="update readme."> - <created>1560273209575</created> - <option name="number" value="00005" /> - <option name="presentableId" value="LOCAL-00005" /> - <option name="project" value="LOCAL" /> - <updated>1560273209575</updated> - </task> - <option name="localTasksCounter" value="6" /> - <servers /> - </component> - <component name="ToolWindowManager"> - <frame x="67" y="25" width="1853" height="1055" extended-state="6" /> - <layout> - <window_info active="true" content_ui="combo" id="Project" order="0" visible="true" weight="0.1568306" /> - <window_info id="Structure" order="1" side_tool="true" weight="0.25" /> - <window_info id="Favorites" order="2" side_tool="true" /> - <window_info anchor="bottom" id="Message" order="0" /> - <window_info anchor="bottom" id="Find" order="1" /> - <window_info anchor="bottom" id="Run" order="2" /> - <window_info anchor="bottom" id="Debug" order="3" weight="0.4" /> - <window_info anchor="bottom" id="Cvs" order="4" weight="0.25" /> - <window_info anchor="bottom" id="Inspection" order="5" weight="0.4" /> - <window_info anchor="bottom" id="TODO" order="6" /> - <window_info anchor="bottom" id="Version Control" order="7" visible="true" weight="0.16306695" /> - <window_info anchor="bottom" id="Terminal" order="8" /> - <window_info anchor="bottom" id="Event Log" order="9" side_tool="true" /> - <window_info anchor="bottom" id="Python Console" order="10" /> - <window_info anchor="right" id="Commander" internal_type="SLIDING" order="0" type="SLIDING" weight="0.4" /> - <window_info anchor="right" id="Ant Build" order="1" weight="0.25" /> - <window_info anchor="right" content_ui="combo" id="Hierarchy" order="2" weight="0.25" /> - </layout> - </component> - <component name="VcsManagerConfiguration"> - <MESSAGE value="correct txt encoding problem." /> - <MESSAGE value="add Singularity Images Folder" /> - <MESSAGE value="update readme" /> - <MESSAGE value="update readme." /> - <option name="LAST_COMMIT_MESSAGE" value="update readme." /> - </component> - <component name="editorHistoryManager"> - <entry file="file://$PROJECT_DIR$/requirements"> - <provider selected="true" editor-type-id="text-editor"> - <state relative-caret-position="180"> - <caret line="9" lean-forward="true" selection-start-line="9" selection-end-line="9" /> - </state> - </provider> - </entry> - <entry file="file://$PROJECT_DIR$/SingularityImages/README.md"> - <provider selected="true" editor-type-id="split-provider[text-editor;markdown-preview-editor]"> - <state split_layout="SPLIT"> - <first_editor relative-caret-position="20"> - <caret line="1" column="27" selection-start-line="1" selection-start-column="27" selection-end-line="1" selection-end-column="27" /> - </first_editor> - <second_editor /> - </state> - </provider> - </entry> - <entry file="file://$PROJECT_DIR$/evalation.py"> - <provider selected="true" editor-type-id="text-editor"> - <state relative-caret-position="2500"> - <caret line="132" column="69" selection-start-line="132" selection-start-column="69" selection-end-line="132" selection-end-column="69" /> - <folding> - <element signature="e#424#433#0" expanded="true" /> - </folding> - </state> - </provider> - </entry> - <entry file="file://$PROJECT_DIR$/utils/tracking_config.py"> - <provider selected="true" editor-type-id="text-editor"> - <state relative-caret-position="240"> - <caret line="12" column="4" selection-start-line="12" selection-start-column="4" selection-end-line="12" selection-end-column="4" /> - </state> - </provider> - </entry> - <entry file="file://$PROJECT_DIR$/plot_results.py"> - <provider selected="true" editor-type-id="text-editor" /> - </entry> - <entry file="file://$PROJECT_DIR$/train_mot.py"> - <provider selected="true" editor-type-id="text-editor"> - <state relative-caret-position="294"> - <caret line="283" column="44" selection-start-line="283" selection-start-column="44" selection-end-line="283" selection-end-column="44" /> - </state> - </provider> - </entry> - <entry file="file://$PROJECT_DIR$/README.md"> - <provider selected="true" editor-type-id="split-provider[text-editor;markdown-preview-editor]"> - <state split_layout="SPLIT"> - <first_editor relative-caret-position="120"> - <caret line="126" column="3" selection-start-line="126" selection-start-column="3" selection-end-line="126" selection-end-column="3" /> - </first_editor> - <second_editor /> - </state> - </provider> - </entry> - <entry file="file://$PROJECT_DIR$/tracking_on_mot.py"> - <provider selected="true" editor-type-id="text-editor"> - <state relative-caret-position="63"> - <caret line="356" lean-forward="true" selection-start-line="356" selection-end-line="356" /> - <folding> - <element signature="e#423#432#0" expanded="true" /> - </folding> - </state> - </provider> - </entry> - </component> -</project> \ No newline at end of file diff --git a/__init__.py b/__init__.py index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..99cc951b67c17ac5e091806ad6bc329426861f7b 100644 --- a/__init__.py +++ b/__init__.py @@ -0,0 +1,9 @@ +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). +# This file is a part of implementation for paper: +# How To Train Your Deep Multi-Object Tracker. +# This contribution is headed by Perception research team, INRIA. +# Contributor(s) : Yihong Xu +# INRIA contact : yihong.xu@inria.fr +# created on 16th April 2020. +#========================================================================== diff --git a/evalation.py b/evalation.py index 69b593377388d20cc14cadbffca847bf28187615..63858fc5c69fc388c2d9962696bef869f4002115 100644 --- a/evalation.py +++ b/evalation.py @@ -1,13 +1,12 @@ -# ========================================================================== -# +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). # This file is a part of implementation for paper: -# DeepMOT: A Differentiable Framework for Training Multiple Object Trackers. +# How To Train Your Deep Multi-Object Tracker. # This contribution is headed by Perception research team, INRIA. -# # Contributor(s) : Yihong Xu # INRIA contact : yihong.xu@inria.fr -# -# =========================================================================== +# created on 16th April 2020. +#========================================================================== import os diff --git a/models/DAN.py b/models/DAN.py index 29b0e9005b3fb1ec173ddc70551e8f49d24c5f4b..5cd876bb31b073d2a049092c493147c1accb5391 100644 --- a/models/DAN.py +++ b/models/DAN.py @@ -1,3 +1,13 @@ +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). +# This file is a part of implementation for paper: +# How To Train Your Deep Multi-Object Tracker. +# This contribution is headed by Perception research team, INRIA. +# Contributor(s) : Yihong Xu +# INRIA contact : yihong.xu@inria.fr +# created on 16th April 2020. +#========================================================================== + """ SST tracker net diff --git a/models/DHN.py b/models/DHN.py index 347769a97dd08c32fb2c88416bb88889eda72da9..52c62c39f8b08d41400d344459e2b52534c8c2cb 100644 --- a/models/DHN.py +++ b/models/DHN.py @@ -1,13 +1,12 @@ -# ========================================================================== -# +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). # This file is a part of implementation for paper: -# DeepMOT: A Differentiable Framework for Training Multiple Object Trackers. +# How To Train Your Deep Multi-Object Tracker. # This contribution is headed by Perception research team, INRIA. -# # Contributor(s) : Yihong Xu # INRIA contact : yihong.xu@inria.fr -# -# =========================================================================== +# created on 16th April 2020. +#========================================================================== import torch import torch.nn as nn diff --git a/models/__init__.py b/models/__init__.py index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..533d5c5ebae8032353d7c0a4a498ab7e9cc31649 100644 --- a/models/__init__.py +++ b/models/__init__.py @@ -0,0 +1,9 @@ +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). +# This file is a part of implementation for paper: +# How To Train Your Deep Multi-Object Tracker. +# This contribution is headed by Perception research team, INRIA. +# Contributor(s) : Yihong Xu +# INRIA contact : yihong.xu@inria.fr +# created on 16th April 2020. +#========================================================================== \ No newline at end of file diff --git a/models/siamrpn.py b/models/siamrpn.py index 853a68d684e9e6e4d6bf7105ea593044c6df878a..e3f09f325b1d66feb6c9f2eaf844c825818c8ce5 100644 --- a/models/siamrpn.py +++ b/models/siamrpn.py @@ -1,13 +1,12 @@ -# ========================================================================== -# +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). # This file is a part of implementation for paper: -# DeepMOT: A Differentiable Framework for Training Multiple Object Trackers. +# How To Train Your Deep Multi-Object Tracker. # This contribution is headed by Perception research team, INRIA. -# # Contributor(s) : Yihong Xu # INRIA contact : yihong.xu@inria.fr -# -# =========================================================================== +# created on 16th April 2020. +#========================================================================== import torch.nn as nn import torch.nn.functional as F diff --git a/plot_results.py b/plot_results.py index bceef3d39b963f9a7413d524d5497dfbed5d5da1..d66c547d59b1e75c4b2bd80ddd2294bd9cb01b66 100644 --- a/plot_results.py +++ b/plot_results.py @@ -1,13 +1,12 @@ -# ========================================================================== -# +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). # This file is a part of implementation for paper: -# DeepMOT: A Differentiable Framework for Training Multiple Object Trackers. +# How To Train Your Deep Multi-Object Tracker. # This contribution is headed by Perception research team, INRIA. -# -# Contributor(s) : Yutong Ban -# INRIA contact : yutong.ban@inria.fr -# -# =========================================================================== +# Contributor(s) : Yihong Xu +# INRIA contact : yihong.xu@inria.fr +# created on 16th April 2020. +#========================================================================== import os import cv2 diff --git a/tracking_on_mot.py b/tracking_on_mot.py index 7fc87a6d3af8e956dfb0008280426f691b381a1f..e1cd14f3c6abab3a03b22f31f9670504ff423e5c 100644 --- a/tracking_on_mot.py +++ b/tracking_on_mot.py @@ -1,13 +1,12 @@ -# ========================================================================== -# +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). # This file is a part of implementation for paper: -# DeepMOT: A Differentiable Framework for Training Multiple Object Trackers. +# How To Train Your Deep Multi-Object Tracker. # This contribution is headed by Perception research team, INRIA. -# # Contributor(s) : Yihong Xu # INRIA contact : yihong.xu@inria.fr -# -# =========================================================================== +# created on 16th April 2020. +#========================================================================== import os import csv diff --git a/train_mot.py b/train_mot.py index 8f3b69bc0fc04b2864858c4341d4bfdc7fd3e212..9777adf8e032274d884b2d272ca8e6e81f0ed087 100644 --- a/train_mot.py +++ b/train_mot.py @@ -1,14 +1,12 @@ - -# ========================================================================== -# +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). # This file is a part of implementation for paper: -# DeepMOT: A Differentiable Framework for Training Multiple Object Trackers. +# How To Train Your Deep Multi-Object Tracker. # This contribution is headed by Perception research team, INRIA. -# # Contributor(s) : Yihong Xu # INRIA contact : yihong.xu@inria.fr -# -# =========================================================================== +# created on 16th April 2020. +#========================================================================== import os import random diff --git a/utils/DAN_utils.py b/utils/DAN_utils.py index ca1964dc787674a83d5908e15b282f7cd4b91cb5..1e59eacc9f6c59573f6124a2f3ac77581153e976 100644 --- a/utils/DAN_utils.py +++ b/utils/DAN_utils.py @@ -1,3 +1,12 @@ +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). +# This file is a part of implementation for paper: +# How To Train Your Deep Multi-Object Tracker. +# This contribution is headed by Perception research team, INRIA. +# Contributor(s) : Yihong Xu +# INRIA contact : yihong.xu@inria.fr +# created on 16th April 2020. +#========================================================================== import cv2 import numpy as np diff --git a/utils/__init__.py b/utils/__init__.py index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..99cc951b67c17ac5e091806ad6bc329426861f7b 100644 --- a/utils/__init__.py +++ b/utils/__init__.py @@ -0,0 +1,9 @@ +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). +# This file is a part of implementation for paper: +# How To Train Your Deep Multi-Object Tracker. +# This contribution is headed by Perception research team, INRIA. +# Contributor(s) : Yihong Xu +# INRIA contact : yihong.xu@inria.fr +# created on 16th April 2020. +#========================================================================== diff --git a/utils/box_utils.py b/utils/box_utils.py index 08f503fec4e32c136aa9af881121a56be2093488..dc0e6e87889043cad73cd748d7130036d967cc3b 100644 --- a/utils/box_utils.py +++ b/utils/box_utils.py @@ -1,13 +1,12 @@ -# ========================================================================== -# +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). # This file is a part of implementation for paper: -# DeepMOT: A Differentiable Framework for Training Multiple Object Trackers. +# How To Train Your Deep Multi-Object Tracker. # This contribution is headed by Perception research team, INRIA. -# # Contributor(s) : Yihong Xu # INRIA contact : yihong.xu@inria.fr -# -# =========================================================================== +# created on 16th April 2020. +#========================================================================== import numpy as np import copy diff --git a/utils/io_utils.py b/utils/io_utils.py index 2172e4e0367e3a39b04120233792cb9dcceb7c59..0bdcaa5d27f2f68ec1b43c353f21de1f9e99aa42 100644 --- a/utils/io_utils.py +++ b/utils/io_utils.py @@ -1,13 +1,12 @@ -# ========================================================================== -# +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). # This file is a part of implementation for paper: -# DeepMOT: A Differentiable Framework for Training Multiple Object Trackers. +# How To Train Your Deep Multi-Object Tracker. # This contribution is headed by Perception research team, INRIA. -# # Contributor(s) : Yihong Xu # INRIA contact : yihong.xu@inria.fr -# -# =========================================================================== +# created on 16th April 2020. +#========================================================================== import csv from utils.box_utils import xywh2xyxy diff --git a/utils/loss.py b/utils/loss.py index 388c14ffb23028856b25b34b2dc68f92dc63b26f..5d538a2974ed8216d847554c403402dfe17d84e8 100644 --- a/utils/loss.py +++ b/utils/loss.py @@ -1,13 +1,12 @@ -# ========================================================================== -# +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). # This file is a part of implementation for paper: -# DeepMOT: A Differentiable Framework for Training Multiple Object Trackers. +# How To Train Your Deep Multi-Object Tracker. # This contribution is headed by Perception research team, INRIA. -# # Contributor(s) : Yihong Xu # INRIA contact : yihong.xu@inria.fr -# -# =========================================================================== +# created on 16th April 2020. +#========================================================================== from utils.box_utils import * import torch.nn.functional as F diff --git a/utils/mot_utils.py b/utils/mot_utils.py index ba9ea037d9dc44f7753e17189c86328c8b2d5cbe..be2164148d7e797ce934f1986704c6114e2da6ba 100644 --- a/utils/mot_utils.py +++ b/utils/mot_utils.py @@ -1,13 +1,12 @@ -# ========================================================================== -# +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). # This file is a part of implementation for paper: -# DeepMOT: A Differentiable Framework for Training Multiple Object Trackers. +# How To Train Your Deep Multi-Object Tracker. # This contribution is headed by Perception research team, INRIA. -# # Contributor(s) : Yihong Xu # INRIA contact : yihong.xu@inria.fr -# -# =========================================================================== +# created on 16th April 2020. +#========================================================================== from utils.sot_utils import * from utils.DAN_utils import TrackUtil @@ -107,7 +106,7 @@ def easy_birth_deathV4_rpn(munkres, bbox_track, det_boxes, curr_img, id_track, c pre_id=None): """ birth and death process during training - :param munkres: assignment matrix {0,1} for track_detection association, numpy array, [batch, H, W] + :param munkres: assignment matrix {0�?} for track_detection association, numpy array, [batch, H, W] :param bbox_track: current tracks bboxes, torch Float tensor, [num_track, 4] :param curr_img: current frame, numpy array, [h,w,c] :param det_boxes: current detection (or gt) boxes [[id, bbox], [id, bbox]], list, [num_boxes, 5] diff --git a/utils/noise_utils.py b/utils/noise_utils.py index b17065be2c446d4d62c57dd28afead20ba229030..c3f6771a6229411d5966b694209b359a9b343744 100644 --- a/utils/noise_utils.py +++ b/utils/noise_utils.py @@ -1,13 +1,12 @@ -# ========================================================================== -# +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). # This file is a part of implementation for paper: -# DeepMOT: A Differentiable Framework for Training Multiple Object Trackers. +# How To Train Your Deep Multi-Object Tracker. # This contribution is headed by Perception research team, INRIA. -# # Contributor(s) : Yihong Xu # INRIA contact : yihong.xu@inria.fr -# -# =========================================================================== +# created on 16th April 2020. +#========================================================================== import numpy as np import cv2 diff --git a/utils/sot_utils.py b/utils/sot_utils.py index 5ad5c03381471078ec628ee3efc1fbda7812f462..baab87377603351061194b3e834510a4eebf1cb8 100644 --- a/utils/sot_utils.py +++ b/utils/sot_utils.py @@ -1,13 +1,12 @@ -# ========================================================================== -# +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). # This file is a part of implementation for paper: -# DeepMOT: A Differentiable Framework for Training Multiple Object Trackers. +# How To Train Your Deep Multi-Object Tracker. # This contribution is headed by Perception research team, INRIA. -# # Contributor(s) : Yihong Xu # INRIA contact : yihong.xu@inria.fr -# -# =========================================================================== +# created on 16th April 2020. +#========================================================================== import torch.nn.functional as F import torch @@ -441,7 +440,7 @@ def tracker_eval(net, x_crop, target_pos, target_sz, window, scale_z, p, templat def update_target_image_train(predicted, id_track, dets, img_curr, states, SOT_tracker, prev_id=None): """ - :param predicted: assignment matrix {0,1} for track_detection association, numpy array, [batch, H, W] + :param predicted: assignment matrix {0�?} for track_detection association, numpy array, [batch, H, W] :param id_track: current tracking ids, list :param dets: current detection (or gt) boxes [[id, bbox], [id, bbox]], list, [num_boxes, 5] :param img_curr: current frame, numpy array, [h,w,c] diff --git a/utils/tracking_config.py b/utils/tracking_config.py index d69e692b0f90a23dda3058a3336c9a300b933c97..673239ea87ba4ea91005a921b084f27ea7412928 100644 --- a/utils/tracking_config.py +++ b/utils/tracking_config.py @@ -1,13 +1,12 @@ -# ========================================================================== -# +#========================================================================== +# This file is under License LGPL-3.0 (see details in the license file). # This file is a part of implementation for paper: -# DeepMOT: A Differentiable Framework for Training Multiple Object Trackers. +# How To Train Your Deep Multi-Object Tracker. # This contribution is headed by Perception research team, INRIA. -# # Contributor(s) : Yihong Xu # INRIA contact : yihong.xu@inria.fr -# -# =========================================================================== +# created on 16th April 2020. +#========================================================================== def tracking_config(vname, dataset):