diff --git a/Dockerfile b/Dockerfile index 53c4b51..fb3e59c 100644 --- a/Dockerfile +++ b/Dockerfile @@ -19,7 +19,8 @@ COPY dt_backend.py ./ COPY dt_frontend.py ./ COPY init_learning_factory_from_cypher_file.py ./ COPY learning_factory_continuous_ordering.py ./ -COPY similarity* ./ +COPY similarity_pipeline similarity_pipeline +COPY mmdetection mmdetection EXPOSE 8050 EXPOSE 8000 diff --git a/testmodel.py b/testmodel.py deleted file mode 100644 index bed83df..0000000 --- a/testmodel.py +++ /dev/null @@ -1,15 +0,0 @@ -_base_ = "./faster_rcnn_r50_fpn_1x_coco.py" -model = dict( - backbone=dict( - type="ResNeXt", - depth=101, - groups=64, - base_width=4, - num_stages=4, - out_indices=(0, 1, 2, 3), - frozen_stages=1, - norm_cfg=dict(type="BN", requires_grad=True), - style="pytorch", - init_cfg=dict(type="Pretrained", checkpoint="open-mmlab://resnext101_64x4d"), - ) -)