NoelShin
commited on
Commit
·
cff2e60
1
Parent(s):
82d5d16
fix import error
Browse files- .idea/workspace.xml +4 -2
- networks/modeling.py +3 -3
- requirements.txt +0 -2
.idea/workspace.xml
CHANGED
|
@@ -5,7 +5,9 @@
|
|
| 5 |
</component>
|
| 6 |
<component name="ChangeListManager">
|
| 7 |
<list default="true" id="9fb9e207-fc4f-4ff3-9adc-3c4c1e67daa7" name="Changes" comment="">
|
| 8 |
-
<change beforePath="$PROJECT_DIR
|
|
|
|
|
|
|
| 9 |
</list>
|
| 10 |
<option name="SHOW_DIALOG" value="false" />
|
| 11 |
<option name="HIGHLIGHT_CONFLICTS" value="true" />
|
|
@@ -47,7 +49,7 @@
|
|
| 47 |
<option name="presentableId" value="Default" />
|
| 48 |
<updated>1664204268713</updated>
|
| 49 |
<workItem from="1664204270261" duration="37000" />
|
| 50 |
-
<workItem from="1664204316867" duration="
|
| 51 |
</task>
|
| 52 |
<servers />
|
| 53 |
</component>
|
|
|
|
| 5 |
</component>
|
| 6 |
<component name="ChangeListManager">
|
| 7 |
<list default="true" id="9fb9e207-fc4f-4ff3-9adc-3c4c1e67daa7" name="Changes" comment="">
|
| 8 |
+
<change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
|
| 9 |
+
<change beforePath="$PROJECT_DIR$/networks/modeling.py" beforeDir="false" afterPath="$PROJECT_DIR$/networks/modeling.py" afterDir="false" />
|
| 10 |
+
<change beforePath="$PROJECT_DIR$/requirements.txt" beforeDir="false" afterPath="$PROJECT_DIR$/requirements.txt" afterDir="false" />
|
| 11 |
</list>
|
| 12 |
<option name="SHOW_DIALOG" value="false" />
|
| 13 |
<option name="HIGHLIGHT_CONFLICTS" value="true" />
|
|
|
|
| 49 |
<option name="presentableId" value="Default" />
|
| 50 |
<updated>1664204268713</updated>
|
| 51 |
<workItem from="1664204270261" duration="37000" />
|
| 52 |
+
<workItem from="1664204316867" duration="4698000" />
|
| 53 |
</task>
|
| 54 |
<servers />
|
| 55 |
</component>
|
networks/modeling.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
-
from networks.
|
| 2 |
-
from networks.
|
| 3 |
-
from networks.
|
| 4 |
|
| 5 |
|
| 6 |
def _segm_hrnet(name, backbone_name, num_classes, pretrained_backbone):
|
|
|
|
| 1 |
+
from networks.utils import IntermediateLayerGetter
|
| 2 |
+
from networks._deeplab import DeepLabHead, DeepLabHeadV3Plus, DeepLabV3
|
| 3 |
+
from networks.backbone import resnet, mobilenetv2, hrnetv2
|
| 4 |
|
| 5 |
|
| 6 |
def _segm_hrnet(name, backbone_name, num_classes, pretrained_backbone):
|
requirements.txt
CHANGED
|
@@ -1,5 +1,3 @@
|
|
| 1 |
opencv-contrib-python==4.5.5.62
|
| 2 |
torch==1.11.0
|
| 3 |
torchvision==0.12.0
|
| 4 |
-
timm==0.4.12
|
| 5 |
-
scipy==1.6.2
|
|
|
|
| 1 |
opencv-contrib-python==4.5.5.62
|
| 2 |
torch==1.11.0
|
| 3 |
torchvision==0.12.0
|
|
|
|
|
|