File tree 2 files changed +17
-2
lines changed
CrossPseudoSupervision/batch_transforms
PanopticSeg/paddlepanseg/datasets
2 files changed +17
-2
lines changed Original file line number Diff line number Diff line change 12
12
# See the License for the specific language governing permissions and
13
13
# limitations under the License.
14
14
15
- from paddle .fluid .dataloader .collate import default_collate_fn
15
+ import paddle
16
+ paddle_version = paddle .__version__ [:3 ]
17
+ # paddle version < 2.5.0 and not develop
18
+ if paddle_version not in ["2.5" , "0.0" ]:
19
+ from paddle .fluid .dataloader .collate import default_collate_fn
20
+ # paddle version >= 2.5.0 or develop
21
+ else :
22
+ from paddle .io .dataloader .collate import default_collate_fn
16
23
17
24
18
25
class SegCollate (object ):
Original file line number Diff line number Diff line change 18
18
19
19
import numpy as np
20
20
import paddle
21
- from paddle . fluid . dataloader . collate import default_collate_fn
21
+
22
22
from paddleseg .transforms import Compose
23
23
24
24
from paddlepanseg .cvlibs import build_info_dict
25
25
26
+ paddle_version = paddle .__version__ [:3 ]
27
+ # paddle version < 2.5.0 and not develop
28
+ if paddle_version not in ["2.5" , "0.0" ]:
29
+ from paddle .fluid .dataloader .collate import default_collate_fn
30
+ # paddle version >= 2.5.0 or develop
31
+ else :
32
+ from paddle .io .dataloader .collate import default_collate_fn
33
+
26
34
27
35
class PanopticDataset (paddle .io .Dataset ):
28
36
NO_COLLATION_KEYS = ('gt_fields' , 'trans_info' , 'image_id' )
You can’t perform that action at this time.
0 commit comments