init

docker/Dockerfile

+FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-23.04-py37-latest
+ENV DEBIAN_FRONTEND=noninteractive
+# COPY requirements.txt requirements.txt
+# RUN pip3 install -r requirements.txt -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com

exps/default/unicorn_det_convnext_large_800x1280.py

+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2022 ByteDance. All Rights Reserved.
+import os
+from unicorn.exp import ExpDet
+
+class Exp(ExpDet):
+    def __init__(self):
+        super(Exp, self).__init__()
+        self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0]
+        self.backbone_name = "convnext_large"
+        self.pretrained_name = "convnext_large_22k_224.pth"
+        self.in_channels = [384, 768, 1536]
+        self.use_checkpoint = True
+        self.input_size = (800, 1280)
+        self.test_size = (800, 1280)
\ No newline at end of file

exps/default/unicorn_det_convnext_tiny_800x1280.py

+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2022 ByteDance. All Rights Reserved.
+import os
+from unicorn.exp import ExpDet
+
+class Exp(ExpDet):
+    def __init__(self):
+        super(Exp, self).__init__()
+        self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0]
+        self.input_size = (800, 1280)
+        self.test_size = (800, 1280)
+        
\ No newline at end of file

exps/default/unicorn_det_r50_800x1280.py

+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2022 ByteDance. All Rights Reserved.
+import os
+from unicorn.exp import ExpDet
+
+class Exp(ExpDet):
+    def __init__(self):
+        super(Exp, self).__init__()
+        self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0]
+        self.backbone_name = "resnet50"
+        self.in_channels = [512, 1024, 2048]
+        self.input_size = (800, 1280)
+        self.test_size = (800, 1280)

exps/default/unicorn_inst_convnext_tiny_800x1280.py

+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2022 ByteDance. All Rights Reserved.
+import os
+from unicorn.exp import ExpDetMask
+
+class Exp(ExpDetMask):
+    def __init__(self):
+        super(Exp, self).__init__()
+        self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0]
+        self.input_size = (800, 1280)
+        self.test_size = (800, 1280)
+        
\ No newline at end of file

exps/default/unicorn_track_large.py

+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2022 ByteDance. All Rights Reserved.
+import os
+from unicorn.exp import ExpTrack
+"""
+The main setting used in the Unicorn paper (ConvNext-Large Backbone)
+We load weights pretrained on COCO with input resolution of 800x1280
+"""
+class Exp(ExpTrack):
+    def __init__(self):
+        super(Exp, self).__init__()
+        self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0]
+        self.backbone_name = "convnext_large"
+        self.in_channels = [384, 768, 1536]
+        self.pretrain_name = "unicorn_det_convnext_large_800x1280"

exps/default/unicorn_track_large_mask.py

+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2022 ByteDance. All Rights Reserved.
+import os
+from unicorn.exp import ExpTrackMask
+"""
+The main setting used in the Unicorn paper (ConvNext-Large Backbone)
+We load weights pretrained on COCO with input resolution of 800x1280
+"""
+class Exp(ExpTrackMask):
+    def __init__(self):
+        super(Exp, self).__init__()
+        self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0]
+        self.backbone_name = "convnext_large"
+        self.in_channels = [384, 768, 1536]
+        self.pretrain_name = "unicorn_track_large"

exps/default/unicorn_track_large_mot_challenge.py

+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2022 ByteDance. All Rights Reserved.
+import os
+from unicorn.exp import ExpTrack
+"""
+The main setting used in the Unicorn paper (ConvNext-Large Backbone)
+We load weights pretrained on COCO with input resolution of 800x1280
+"""
+class Exp(ExpTrack):
+    def __init__(self):
+        super(Exp, self).__init__()
+        self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0]
+        self.backbone_name = "convnext_large"
+        self.in_channels = [384, 768, 1536]
+        self.pretrain_name = "unicorn_det_convnext_large_800x1280"
+        self.mot_test_name = "motchallenge"
+        self.num_classes = 1
+        self.mhs = False

exps/default/unicorn_track_large_mot_challenge_mask.py

+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2022 ByteDance. All Rights Reserved.
+import os
+from unicorn.exp import ExpTrackMask
+"""
+The main setting used in the Unicorn paper (ConvNext-Large Backbone)
+We load weights pretrained on COCO with input resolution of 800x1280
+"""
+class Exp(ExpTrackMask):
+    def __init__(self):
+        super(Exp, self).__init__()
+        self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0]
+        self.backbone_name = "convnext_large"
+        self.in_channels = [384, 768, 1536]
+        self.pretrain_name = "unicorn_track_large_mot_challenge"
+        self.mot_test_name = "motchallenge"
+        self.num_classes = 1
+        self.mhs = False

exps/default/unicorn_track_r50.py

+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2022 ByteDance. All Rights Reserved.
+import os
+from unicorn.exp import ExpTrack
+"""
+The R50 setting used in the ablation
+We load weights pretrained on COCO with input resolution of 800x1280
+"""
+class Exp(ExpTrack):
+    def __init__(self):
+        super(Exp, self).__init__()
+        self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0]
+        self.backbone_name = "resnet50"
+        self.in_channels = [512, 1024, 2048]
+        self.pretrain_name = "unicorn_det_r50_800x1280"
+        
+

exps/default/unicorn_track_r50_mask.py

+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2022 ByteDance. All Rights Reserved.
+import os
+from unicorn.exp import ExpTrackMask
+"""
+The R50 setting used in the ablation
+We load weights pretrained on COCO with input resolution of 800x1280
+"""
+class Exp(ExpTrackMask):
+    def __init__(self):
+        super(Exp, self).__init__()
+        self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0]
+        self.backbone_name = "resnet50"
+        self.in_channels = [512, 1024, 2048]
+        self.pretrain_name = "unicorn_track_r50"
+        
+

exps/default/unicorn_track_tiny.py

+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2022 ByteDance. All Rights Reserved.
+import os
+from unicorn.exp import ExpTrack
+"""
+The baseline setting used in the ablation
+We load weights pretrained on COCO with input resolution of 800x1280
+"""
+class Exp(ExpTrack):
+    def __init__(self):
+        super(Exp, self).__init__()
+        self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0]
+        self.pretrain_name = "unicorn_det_convnext_tiny_800x1280"
+        
+

exps/default/unicorn_track_tiny_mask.py

+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# Copyright (c) 2022 ByteDance. All Rights Reserved.
+import os
+from unicorn.exp import ExpTrackMask
+"""
+The baseline setting used in the ablation
+We load weights pretrained on COCO with input resolution of 800x1280
+"""
+class Exp(ExpTrackMask):
+    def __init__(self):
+        super(Exp, self).__init__()
+        self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0]
+        self.pretrain_name = "unicorn_track_tiny"
+        
+